diff --git a/04_mnist_basics.ipynb b/04_mnist_basics.ipynb
index 675bb5b..98be6a3 100644
--- a/04_mnist_basics.ipynb
+++ b/04_mnist_basics.ipynb
@@ -1756,7 +1756,7 @@
     }
    ],
    "source": [
-    "F.l1_loss(a_3.float(),mean7), F.mse_loss(a_3,mean7).sqrt()"
+    "F.l1_loss(a_3,mean7), F.mse_loss(a_3,mean7).sqrt()"
    ]
   },
   {
diff --git a/clean/04_mnist_basics.ipynb b/clean/04_mnist_basics.ipynb
index a75eb39..bcde7fc 100644
--- a/clean/04_mnist_basics.ipynb
+++ b/clean/04_mnist_basics.ipynb
@@ -255,7 +255,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "F.l1_loss(a_3.float(),mean7), F.mse_loss(a_3,mean7).sqrt()"
+    "F.l1_loss(a_3,mean7), F.mse_loss(a_3,mean7).sqrt()"
    ]
   },
   {