diff --git a/python/src/diffusionkit/mlx/__init__.py b/python/src/diffusionkit/mlx/__init__.py
index 6c6499a..3d41ab3 100644
--- a/python/src/diffusionkit/mlx/__init__.py
+++ b/python/src/diffusionkit/mlx/__init__.py
@@ -256,7 +256,7 @@ def denoise_latents(
         pooled_conditioning,
         num_steps: int = 2,
         cfg_weight: float = 0.0,
-        latent_size: Tuple[int] = (64, 64),
+        latent_size: Tuple[int, int] = (64, 64),
         seed=None,
         image_path: Optional[str] = None,
         denoise: float = 1.0,
@@ -297,7 +297,7 @@ def generate_image(
         num_steps: int = 2,
         cfg_weight: float = 0.0,
         negative_text: str = "",
-        latent_size: Tuple[int] = (64, 64),
+        latent_size: Tuple[int, int] = (64, 64),
         seed=None,
         verbose: bool = True,
         image_path: Optional[str] = None,