diff --git a/main_lightning.ipynb b/main_lightning.ipynb
index bbc59a9655646bcfd9a18ce11b08a0c405e86dd2..31cb0c90e9e2a9461e5e04fadca41e40544c97d6 100644
--- a/main_lightning.ipynb
+++ b/main_lightning.ipynb
@@ -513,7 +513,7 @@
    "source": [
     "use_lightning = True\n",
     "use_fp16 = True\n",
-    "mixed_precision=16"
+    "mixed_precision=\"16-mixed\""
    ]
   },
   {