diff --git a/omx/gstomxvideoenc.c b/omx/gstomxvideoenc.c
index d3e7676cfb..b81c919007 100644
--- a/omx/gstomxvideoenc.c
+++ b/omx/gstomxvideoenc.c
@@ -1275,7 +1275,7 @@ gst_omx_video_enc_set_format (GstVideoEncoder * encoder,
 
   if (self->input_state)
     gst_video_codec_state_unref (self->input_state);
-  self->input_state = gst_video_codec_state_ref (self->input_state);
+  self->input_state = gst_video_codec_state_ref (state);
 
   /* Start the srcpad loop again */
   GST_DEBUG_OBJECT (self, "Starting task again");