diff --git a/drivers/video/rockchip/hdmi/rk_hdmi_task.c b/drivers/video/rockchip/hdmi/rk_hdmi_task.c
index 26fffd439910..9674407bea98 100755
--- a/drivers/video/rockchip/hdmi/rk_hdmi_task.c
+++ b/drivers/video/rockchip/hdmi/rk_hdmi_task.c
@@ -81,8 +81,8 @@ void hdmi_sys_remove(struct hdmi *hdmi)
 	INIT_LIST_HEAD(&hdmi->edid.modelist);
 	hdmi->display = HDMI_DISABLE;
 	if (hdmi->set_vif)
-		hdmi->set_vif(hdmi, hdmi->lcdc->cur_screen, 0);
-	rk_fb_switch_screen(hdmi->lcdc->cur_screen, 0, hdmi->lcdc->id);
+		hdmi->set_vif(hdmi, hdmi->lcdc->screen1, 0);
+	rk_fb_switch_screen(hdmi->lcdc->screen1, 0, hdmi->lcdc->id);
 	kobject_uevent_env(&hdmi->ddev->dev->kobj, KOBJ_REMOVE, envp);
 
 #ifdef CONFIG_SWITCH