diff --git a/control/latent_keyframe_nodes.py b/control/latent_keyframe_nodes.py
index 87c820c..78f5527 100644
--- a/control/latent_keyframe_nodes.py
+++ b/control/latent_keyframe_nodes.py
@@ -162,7 +162,7 @@ class LatentKeyframeInterpolationNodeImport:
                         strength_to: float,
                         interpolation: str,
                         revert_direction_at_midpoint: bool=False,
-                        prev_latent_keyframe: LatentKeyframeGroup=None
+                        prev_latent_keyframe: LatentKeyframeGroupImport=None
                         ):