Minor.

OpenSourceEconomics · Feb 7, 2025 · c5d5ac7 · c5d5ac7
1 parent ae84fa4
commit c5d5ac7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/docs/source/Tutorials/two_period_model_tutorial.ipynb b/docs/source/Tutorials/two_period_model_tutorial.ipynb
@@ -621,7 +621,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "state_dict = {\"ltc\": 0, \"lagged_choice\": 0, \"period\": 0, \"choice\": choice_in_period_0}\n",
+    "state_dict = {\n",
+    "    \"ltc\": initial_condition[\"health\"],\n",
+    "    \"lagged_choice\": 0,\n",
+    "    \"period\": 0,\n",
+    "    \"choice\": choice_in_period_0,\n",
+    "}\n",
     "\n",
     "cons_calc, value = policy_and_value_for_state_choice_vec(\n",
     "    endog_grid_solved=endog_grid_solved,\n",