diff --git a/clean/04_mnist_basics.ipynb b/clean/04_mnist_basics.ipynb
index 6cc0bbf..a75eb39 100644
--- a/clean/04_mnist_basics.ipynb
+++ b/clean/04_mnist_basics.ipynb
@@ -174,7 +174,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "stacked_sevens = torch.stack(seven_tensors)/255\n",
+    "stacked_sevens = torch.stack(seven_tensors).float()/255\n",
     "stacked_threes = torch.stack(three_tensors).float()/255\n",
     "stacked_threes.shape"
    ]