diff --git a/spatialpy/core/domain.py b/spatialpy/core/domain.py
index 81ea1f27..f7afc481 100644
--- a/spatialpy/core/domain.py
+++ b/spatialpy/core/domain.py
@@ -1035,7 +1035,8 @@ def plot_types(self, width=None, height=None, colormap=None, size=None, title=No
 
         if return_plotly_figure:
             return fig
-        init_notebook_mode(connected=True)
+        # init_notebook_mode(connected=True)
+        init_notebook_mode()
         iplot(fig)
         return
 
diff --git a/spatialpy/core/lattice.py b/spatialpy/core/lattice.py
index d23ce53d..2e7ca70a 100644
--- a/spatialpy/core/lattice.py
+++ b/spatialpy/core/lattice.py
@@ -548,7 +548,7 @@ def __get_types(self):
 
                     if self.type_ids is not None:
                         type_id = self.type_ids[type_id]
-                    type_ids[int(ndx)] = f"type_{type_id}"
+                    type_ids[int(ndx)] = type_id
                 except ValueError as err:
                     errmsg = f"Could not read in subdomain file, error on line {lnum}: {line}"
                     raise LatticeError(errmsg) from err
@@ -685,7 +685,7 @@ def __get_types(self):
 
                     if self.type_ids is not None:
                         type_id = self.type_ids[type_id]
-                    type_ids[int(ndx)] = f"type_{type_id}"
+                    type_ids[int(ndx)] = type_id
                 except ValueError as err:
                     errmsg = f"Could not read in subdomain file, error on line {lnum}: {line}"
                     raise LatticeError(errmsg) from err