diff --git a/site/en/tutorials/generative/autoencoder.ipynb b/site/en/tutorials/generative/autoencoder.ipynb
index 508b136307f..d81628fb401 100644
--- a/site/en/tutorials/generative/autoencoder.ipynb
+++ b/site/en/tutorials/generative/autoencoder.ipynb
@@ -179,7 +179,7 @@
         "    return decoded\n",
         "\n",
         "\n",
-        "shape = x_test.shape\n",
+        "shape = x_test.shape[1:]\n",
         "latent_dim = 64\n",
         "autoencoder = Autoencoder(latent_dim, shape)\n"
       ]