const struct drm_crtc_helper_funcs *crtc_funcs;
        struct drm_crtc *crtc = old_crtc_state->crtc;
        struct drm_atomic_state *old_state = old_crtc_state->state;
+       struct drm_crtc_state *new_crtc_state =
+               drm_atomic_get_new_crtc_state(old_state, crtc);
        struct drm_plane *plane;
        unsigned plane_mask;
 
        plane_mask = old_crtc_state->plane_mask;
-       plane_mask |= crtc->state->plane_mask;
+       plane_mask |= new_crtc_state->plane_mask;
 
        crtc_funcs = crtc->helper_private;
        if (crtc_funcs && crtc_funcs->atomic_begin)
        drm_for_each_plane_mask(plane, crtc->dev, plane_mask) {
                struct drm_plane_state *old_plane_state =
                        drm_atomic_get_old_plane_state(old_state, plane);
+               struct drm_plane_state *new_plane_state =
+                       drm_atomic_get_new_plane_state(old_state, plane);
                const struct drm_plane_helper_funcs *plane_funcs;
 
                plane_funcs = plane->helper_private;
                if (!old_plane_state || !plane_funcs)
                        continue;
 
-               WARN_ON(plane->state->crtc && plane->state->crtc != crtc);
+               WARN_ON(new_plane_state->crtc &&
+                       new_plane_state->crtc != crtc);
 
-               if (drm_atomic_plane_disabling(old_plane_state, plane->state) &&
+               if (drm_atomic_plane_disabling(old_plane_state, new_plane_state) &&
                    plane_funcs->atomic_disable)
                        plane_funcs->atomic_disable(plane, old_plane_state);
-               else if (plane->state->crtc ||
-                        drm_atomic_plane_disabling(old_plane_state, plane->state))
+               else if (new_plane_state->crtc ||
+                        drm_atomic_plane_disabling(old_plane_state, new_plane_state))
                        plane_funcs->atomic_update(plane, old_plane_state);
        }