struct pv_muxing_priv {
        struct vc4_dev *vc4;
-       struct drm_modeset_acquire_ctx ctx;
        struct drm_atomic_state *state;
 };
 
 static int vc4_pv_muxing_test_init(struct kunit *test)
 {
        const struct pv_muxing_param *params = test->param_value;
+       struct drm_modeset_acquire_ctx *ctx;
        struct drm_atomic_state *state;
        struct pv_muxing_priv *priv;
        struct drm_device *drm;
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, vc4);
        priv->vc4 = vc4;
 
-       drm_modeset_acquire_init(&priv->ctx, 0);
+       ctx = drm_kunit_helper_acquire_ctx_alloc(test);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
 
        drm = &vc4->base;
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &priv->ctx;
+       state->acquire_ctx = ctx;
 
        priv->state = state;
 
        struct drm_atomic_state *state = priv->state;
 
        drm_atomic_state_put(state);
-       drm_modeset_drop_locks(&priv->ctx);
-       drm_modeset_acquire_fini(&priv->ctx);
 }
 
 static struct kunit_case vc4_pv_muxing_tests[] = {
  */
 static void drm_test_vc5_pv_muxing_bugs_subsequent_crtc_enable(struct kunit *test)
 {
-       struct drm_modeset_acquire_ctx ctx;
+       struct drm_modeset_acquire_ctx *ctx;
        struct drm_atomic_state *state;
        struct vc4_crtc_state *new_vc4_crtc_state;
        struct vc4_hvs_state *new_hvs_state;
        vc4 = vc5_mock_device(test);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, vc4);
 
-       drm_modeset_acquire_init(&ctx, 0);
+       ctx = drm_kunit_helper_acquire_ctx_alloc(test);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
 
        drm = &vc4->base;
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_add_output(test, state, VC4_ENCODER_TYPE_HDMI0);
        KUNIT_ASSERT_EQ(test, ret, 0);
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_add_output(test, state, VC4_ENCODER_TYPE_HDMI1);
        KUNIT_ASSERT_EQ(test, ret, 0);
        KUNIT_EXPECT_NE(test, hdmi0_channel, hdmi1_channel);
 
        drm_atomic_state_put(state);
-       drm_modeset_drop_locks(&ctx);
-       drm_modeset_acquire_fini(&ctx);
 }
 
 static void drm_test_vc5_pv_muxing_bugs_stable_fifo(struct kunit *test)
 {
-       struct drm_modeset_acquire_ctx ctx;
+       struct drm_modeset_acquire_ctx *ctx;
        struct drm_atomic_state *state;
        struct vc4_crtc_state *new_vc4_crtc_state;
        struct vc4_hvs_state *new_hvs_state;
        vc4 = vc5_mock_device(test);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, vc4);
 
-       drm_modeset_acquire_init(&ctx, 0);
+       ctx = drm_kunit_helper_acquire_ctx_alloc(test);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
 
        drm = &vc4->base;
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_add_output(test, state, VC4_ENCODER_TYPE_HDMI0);
        KUNIT_ASSERT_EQ(test, ret, 0);
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_del_output(test, state, VC4_ENCODER_TYPE_HDMI0);
        KUNIT_ASSERT_EQ(test, ret, 0);
        }
 
        drm_atomic_state_put(state);
-       drm_modeset_drop_locks(&ctx);
-       drm_modeset_acquire_fini(&ctx);
 }
 
 static void
 drm_test_vc5_pv_muxing_bugs_subsequent_crtc_enable_too_many_crtc_state(struct kunit *test)
 {
-       struct drm_modeset_acquire_ctx ctx;
+       struct drm_modeset_acquire_ctx *ctx;
        struct drm_atomic_state *state;
        struct vc4_crtc_state *new_vc4_crtc_state;
        struct drm_device *drm;
        vc4 = vc5_mock_device(test);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, vc4);
 
-       drm_modeset_acquire_init(&ctx, 0);
+       ctx = drm_kunit_helper_acquire_ctx_alloc(test);
+       KUNIT_ASSERT_NOT_ERR_OR_NULL(test, ctx);
 
        drm = &vc4->base;
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_add_output(test, state, VC4_ENCODER_TYPE_HDMI0);
        KUNIT_ASSERT_EQ(test, ret, 0);
        state = drm_atomic_state_alloc(drm);
        KUNIT_ASSERT_NOT_ERR_OR_NULL(test, state);
 
-       state->acquire_ctx = &ctx;
+       state->acquire_ctx = ctx;
 
        ret = vc4_mock_atomic_add_output(test, state, VC4_ENCODER_TYPE_HDMI1);
        KUNIT_ASSERT_EQ(test, ret, 0);
        KUNIT_EXPECT_NULL(test, new_vc4_crtc_state);
 
        drm_atomic_state_put(state);
-       drm_modeset_drop_locks(&ctx);
-       drm_modeset_acquire_fini(&ctx);
 }
 
 static struct kunit_case vc5_pv_muxing_bugs_tests[] = {