diff --git a/mmd_vae_pytorchver.ipynb b/mmd_vae_pytorchver.ipynb
index 02cdc42..4948f21 100644
--- a/mmd_vae_pytorchver.ipynb
+++ b/mmd_vae_pytorchver.ipynb
@@ -438,8 +438,8 @@
     "        \n",
     "    # show images\n",
     "    gen_z = Variable(torch.randn((100, z_dim)))\n",
-    "    if(usecuda):\n"
-    "        gen_z = gen_z.cuda(idgpu)\n"
+    "    if(usecuda):\n",
+    "        gen_z = gen_z.cuda(idgpu)\n",
     "    samples = decodermodel(gen_z)\n",
     "    samples =samples.view(100,28,28,1)\n",
     "    plt.imshow(convert_to_display(samples.data), cmap='Greys_r')\n",