.s_ctrl = imx290_set_ctrl,
 };
 
+static struct v4l2_mbus_framefmt *
+imx290_get_pad_format(struct imx290 *imx290, struct v4l2_subdev_state *state,
+                     u32 which)
+{
+       if (which == V4L2_SUBDEV_FORMAT_ACTIVE)
+               return &imx290->current_format;
+       else
+               return v4l2_subdev_get_try_format(&imx290->sd, state, 0);
+}
+
 static int imx290_enum_mbus_code(struct v4l2_subdev *sd,
                                 struct v4l2_subdev_state *sd_state,
                                 struct v4l2_subdev_mbus_code_enum *code)
 
        mutex_lock(&imx290->lock);
 
-       if (fmt->which == V4L2_SUBDEV_FORMAT_TRY)
-               framefmt = v4l2_subdev_get_try_format(&imx290->sd, sd_state,
-                                                     fmt->pad);
-       else
-               framefmt = &imx290->current_format;
-
+       framefmt = imx290_get_pad_format(imx290, sd_state, fmt->which);
        fmt->format = *framefmt;
 
        mutex_unlock(&imx290->lock);
        fmt->format.code = imx290_formats[i].code;
        fmt->format.field = V4L2_FIELD_NONE;
 
-       if (fmt->which == V4L2_SUBDEV_FORMAT_TRY) {
-               format = v4l2_subdev_get_try_format(sd, sd_state, fmt->pad);
-       } else {
-               format = &imx290->current_format;
+       format = imx290_get_pad_format(imx290, sd_state, fmt->which);
+
+       if (fmt->which == V4L2_SUBDEV_FORMAT_ACTIVE) {
                imx290->current_mode = mode;
                imx290->bpp = imx290_formats[i].bpp;