diff --git a/scvi/data/anndata/fields/_obsm_field.py b/scvi/data/anndata/fields/_obsm_field.py
index 98ffb5db80..287e5a0450 100644
--- a/scvi/data/anndata/fields/_obsm_field.py
+++ b/scvi/data/anndata/fields/_obsm_field.py
@@ -71,7 +71,7 @@ def __init__(
         obsm_key: str,
         colnames_uns_key: Optional[str] = None,
         is_count_data: bool = False,
-        correct_data_format=True,
+        correct_data_format: bool = True,
     ) -> None:
         super().__init__(registry_key)
         self._attr_key = obsm_key