diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py
index 3243cc0fd3..4ab64abc23 100644
--- a/invokeai/app/invocations/latent.py
+++ b/invokeai/app/invocations/latent.py
@@ -518,7 +518,7 @@ class DenoiseLatentsInvocation(BaseInvocation):
     def get_conditioning_data(
         self,
         context: InvocationContext,
-        unet,
+        unet: UNet2DConditionModel,
         latent_height: int,
         latent_width: int,
     ) -> TextConditioningData: