1 files changed, 4 insertions, 4 deletions

diff --git a/drivers/gpu/drm/drm_atomic_helper.c b/drivers/gpu/drm/drm_atomic_helper.c
index 2fa0840694d0..2ee509c92034 100644
--- a/drivers/gpu/drm/drm_atomic_helper.c
+++ b/drivers/gpu/drm/drm_atomic_helper.c
@@ -1222,7 +1222,7 @@ retry:
                 goto fail;
         }
 
-        ret = drm_atomic_set_crtc_for_plane(plane_state, crtc);
+        ret = drm_atomic_set_crtc_for_plane(state, plane, crtc);
         if (ret != 0)
                 goto fail;
         drm_atomic_set_fb_for_plane(plane_state, fb);
@@ -1301,7 +1301,7 @@ retry:
                 goto fail;
         }
 
-        ret = drm_atomic_set_crtc_for_plane(plane_state, NULL);
+        ret = drm_atomic_set_crtc_for_plane(state, plane, NULL);
         if (ret != 0)
                 goto fail;
         drm_atomic_set_fb_for_plane(plane_state, NULL);
@@ -1472,7 +1472,7 @@ retry:
                 goto fail;
         }
 
-        ret = drm_atomic_set_crtc_for_plane(primary_state, crtc);
+        ret = drm_atomic_set_crtc_for_plane(state, crtc->primary, crtc);
         if (ret != 0)
                 goto fail;
         drm_atomic_set_fb_for_plane(primary_state, set->fb);
@@ -1744,7 +1744,7 @@ retry:
                 goto fail;
         }
 
-        ret = drm_atomic_set_crtc_for_plane(plane_state, crtc);
+        ret = drm_atomic_set_crtc_for_plane(state, plane, crtc);
         if (ret != 0)
                 goto fail;
         drm_atomic_set_fb_for_plane(plane_state, fb);