diff --git a/magnum_cluster_api/resources.py b/magnum_cluster_api/resources.py
index df847727..1615bd38 100644
--- a/magnum_cluster_api/resources.py
+++ b/magnum_cluster_api/resources.py
@@ -2288,7 +2288,7 @@ def create_cluster_class(
 
 def mutate_machine_deployment(
     context: context.RequestContext,
-    cluster: objects.Cluster,
+    cluster: magnum_objects.Cluster,
     node_group: magnum_objects.NodeGroup,
     machine_deployment: dict = None,
 ):
@@ -2393,7 +2393,7 @@ def mutate_machine_deployment(
 
 
 def generate_machine_deployments_for_cluster(
-    context: context.RequestContext, cluster: objects.Cluster
+    context: context.RequestContext, cluster: magnum_objects.Cluster
 ) -> list:
     machine_deployments = []
     for ng in cluster.nodegroups: