diff --git a/cities/modeling/simple_linear.py b/cities/modeling/simple_linear.py
index ff9e4105..d28b9a6c 100644
--- a/cities/modeling/simple_linear.py
+++ b/cities/modeling/simple_linear.py
@@ -113,7 +113,7 @@ def forward(
                     ].squeeze(-1)
 
                 objects_cat_weighted[name] = weights_categorical_outcome[name][
-                    ..., categorical[name]
+                        ..., categorical[name]
                 ]
 
             # most likely too add hoc and now redundant
@@ -242,3 +242,20 @@ def unconditioned_model():
 
     def forward(self):
         return self.conditioned_model()
+
+
+
+#TODO mypy linting
+
+# + mypy --ignore-missing-imports cities/
+# cities/modeling/simple_linear.py:26: error: Name "pyro.nn.PyroModule" is not defined  [name-defined]
+# cities/modeling/simple_linear.py:72: error: Module has no attribute "sample"  [attr-defined]
+# cities/modeling/simple_linear.py:74: error: Module has no attribute "plate"  [attr-defined]
+# cities/modeling/simple_linear.py:97: error: Module has no attribute "plate"  [attr-defined]
+# cities/modeling/simple_linear.py:102: error: Module has no attribute "sample"  [attr-defined]
+# cities/modeling/simple_linear.py:143: error: Module has no attribute "plate"  [attr-defined]
+# cities/modeling/simple_linear.py:144: error: Module has no attribute "sample"  [attr-defined]
+# cities/modeling/simple_linear.py:154: error: Module has no attribute "sample"  [attr-defined]
+# cities/modeling/simple_linear.py:176: error: Module has no attribute "deterministic"  [attr-defined]
+# cities/modeling/simple_linear.py:182: error: Module has no attribute "sample"  [attr-defined]
+# cities/modeling/simple_linear.py:191: error: Name "pyro.nn.PyroModule" is not defined  [name-defined]
\ No newline at end of file