diff --git a/drivers/gpu/drm/amd/display/dc/dcn301/dcn301_resource.c b/drivers/gpu/drm/amd/display/dc/dcn301/dcn301_resource.c
index cd69e2d2e46302493fa2bcbd334328fa4a90d8d0..fe46a0b911fcf3dc1ad2e73b6b4757189f6f8be4 100644
--- a/drivers/gpu/drm/amd/display/dc/dcn301/dcn301_resource.c
+++ b/drivers/gpu/drm/amd/display/dc/dcn301/dcn301_resource.c
@@ -1907,7 +1907,7 @@ static bool dcn301_resource_construct(
 	/* ABM (or ABMs for NV2x) */
 	/* TODO: */
 	for (i = 0; i < pool->base.res_cap->num_timing_generator; i++) {
-		pool->base.multiple_abms[i] = dce_abm_create(ctx,
+		pool->base.multiple_abms[i] = dmub_abm_create(ctx,
 				&abm_regs[i],
 				&abm_shift,
 				&abm_mask);