diff --git a/drivers/video/fbdev/msm/mdss_mdp_intf_writeback.c b/drivers/video/fbdev/msm/mdss_mdp_intf_writeback.c
index 7e1a16a6d976..87f30349f345 100644
--- a/drivers/video/fbdev/msm/mdss_mdp_intf_writeback.c
+++ b/drivers/video/fbdev/msm/mdss_mdp_intf_writeback.c
@@ -51,6 +51,7 @@ struct mdss_mdp_writeback_ctx {
 	u32 opmode;
 	struct mdss_mdp_format_params *dst_fmt;
 	u16 img_width;
+	u16 img_height;
 	u16 width;
 	u16 height;
 	struct mdss_rect dst_rect;
@@ -236,7 +237,7 @@ static int mdss_mdp_writeback_format_setup(struct mdss_mdp_writeback_ctx *ctx,
 		return -EINVAL;
 	}
 
-	mdss_mdp_get_plane_sizes(fmt, ctx->img_width, ctx->height,
+	mdss_mdp_get_plane_sizes(fmt, ctx->img_width, ctx->img_height,
 				 &ctx->dst_planes,
 				 ctx->opmode & MDSS_MDP_OP_BWC_EN, rotation);
 
@@ -366,6 +367,7 @@ static int mdss_mdp_writeback_prepare_wfd(struct mdss_mdp_ctl *ctl, void *arg)
 
 	ctx->opmode = 0;
 	ctx->img_width = ctl->width;
+	ctx->img_height = ctl->height;
 	ctx->width = ctl->width;
 	ctx->height = ctl->height;
 	ctx->dst_rect.x = 0;
@@ -421,6 +423,7 @@ static int mdss_mdp_writeback_prepare_rot(struct mdss_mdp_ctl *ctl, void *arg)
 	ctx->opmode |= ctx->bwc_mode;
 
 	ctx->img_width = item->output.width;
+	ctx->img_height = item->output.height;
 	ctx->width = ctx->dst_rect.w = item->dst_rect.w;
 	ctx->height = ctx->dst_rect.h = item->dst_rect.h;
 	ctx->dst_rect.x = item->dst_rect.x;