diff --git a/docs/physics/setup/model.ipynb b/docs/physics/setup/model.ipynb
index 5bd1f3382b4..355121bd4ca 100644
--- a/docs/physics/setup/model.ipynb
+++ b/docs/physics/setup/model.ipynb
@@ -53,7 +53,7 @@
    "source": [
     "# We import the necessary packages\n",
     "from tardis.io.configuration.config_reader import Configuration\n",
-    "from tardis.model import SimulationStatetetete\n",
+    "from tardis.model import SimulationState\n",
     "from tardis.io.atom_data.util import download_atom_data\n",
     "from astropy import units as u\n",
     "import matplotlib.pyplot as plt\n",
@@ -103,7 +103,7 @@
     "shell_config.model.structure.velocity.stop = 2000 * u.km/u.s\n",
     "shell_config.model.structure.velocity.num = 20\n",
     "\n",
-    "shell_model = SimulationStatetetete.from_config(shell_config)\n",
+    "shell_model = SimulationState.from_config(shell_config)\n",
     "\n",
     "print('velocity:\\n', shell_model.velocity)\n",
     "print('v_inner:\\n', shell_model.v_inner)\n",
@@ -181,7 +181,7 @@
     "\n",
     "w7_density_config.model.structure.density.type = 'branch85_w7'\n",
     "\n",
-    "w7_density_model = SimulationStatetetete.from_config(w7_density_config)\n",
+    "w7_density_model = SimulationState.from_config(w7_density_config)\n",
     "\n",
     "print('density:\\n', w7_density_model.density)\n",
     "\n",
@@ -209,7 +209,7 @@
     "\n",
     "w7_modified_config.supernova.time_explosion = 12 * u.day\n",
     "\n",
-    "w7_modified_model = SimulationStatetetete.from_config(w7_modified_config)\n",
+    "w7_modified_model = SimulationState.from_config(w7_modified_config)\n",
     "\n",
     "print('density:\\n', w7_modified_model.density)\n",
     "\n",
@@ -273,7 +273,7 @@
     "uni_density_config.model.structure.density.time_0 = 1 * u.day\n",
     "uni_density_config.model.structure.density.value = 5e-10 * u.kg/u.cm**3\n",
     "\n",
-    "uni_density_model = SimulationStatetetete.from_config(uni_density_config)\n",
+    "uni_density_model = SimulationState.from_config(uni_density_config)\n",
     "\n",
     "print('density:\\n', uni_density_model.density)\n",
     "\n",
@@ -327,7 +327,7 @@
     "pow_density_config.model.structure.density.v_0 = 500 * u.km/u.s\n",
     "pow_density_config.model.structure.density.exponent = -2\n",
     "\n",
-    "pow_density_model = SimulationStatetetete.from_config(pow_density_config)\n",
+    "pow_density_model = SimulationState.from_config(pow_density_config)\n",
     "\n",
     "print('density:\\n', pow_density_model.density)\n",
     "\n",
@@ -378,7 +378,7 @@
     "exp_density_config.model.structure.density.rho_0 = 5e-10 * u.kg/u.cm**3\n",
     "exp_density_config.model.structure.density.v_0 = 500 * u.km/u.s\n",
     "\n",
-    "exp_density_model = SimulationStatetetete.from_config(exp_density_config)\n",
+    "exp_density_model = SimulationState.from_config(exp_density_config)\n",
     "\n",
     "print('density:\\n', exp_density_model.density)\n",
     "\n",
@@ -423,7 +423,7 @@
     "                                            'Si':.3,\n",
     "                                            'He': .1}\n",
     "\n",
-    "abund_model = SimulationStatetetete.from_config(abund_config)\n",
+    "abund_model = SimulationState.from_config(abund_config)\n",
     "\n",
     "abund_model.abundance"
    ]
@@ -459,7 +459,7 @@
     "                                            'Ni57':.1,\n",
     "                                            'Cr51':.4}\n",
     "\n",
-    "abund_isotopes_model = SimulationStatetetete.from_config(abund_isotopes_config)\n",
+    "abund_isotopes_model = SimulationState.from_config(abund_isotopes_config)\n",
     "\n",
     "abund_isotopes_model.abundance"
    ]
@@ -527,7 +527,7 @@
     "t_rad_config.model.structure.velocity.stop = 2000 * u.km/u.s\n",
     "t_rad_config.model.structure.velocity.num = 20\n",
     "\n",
-    "t_rad_model = SimulationStatetetete.from_config(t_rad_config)\n",
+    "t_rad_model = SimulationState.from_config(t_rad_config)\n",
     "\n",
     "print('t_inner:\\n', t_rad_model.t_inner)\n",
     "print('t_rad:\\n', t_rad_model.t_rad)\n",
@@ -590,7 +590,7 @@
     "w_config.model.structure.velocity.stop = 2000 * u.km/u.s\n",
     "w_config.model.structure.velocity.num = 20\n",
     "\n",
-    "w_model = SimulationStatetetete.from_config(w_config)\n",
+    "w_model = SimulationState.from_config(w_config)\n",
     "\n",
     "print('w:\\n', w_model.w)\n",
     "\n",