diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb
index c01b309c..e07845ca 100644
--- a/Disco_Diffusion.ipynb
+++ b/Disco_Diffusion.ipynb
@@ -1050,7 +1050,8 @@
         "                else:\n",
         "                    TF.to_pil_image(cutouts[-1].clamp(0, 1).squeeze(0)).save(\"cutout_InnerCrop.jpg\",quality=99)\n",
         "        cutouts = torch.cat(cutouts)\n",
-        "        if skip_augs is not True: cutouts=self.augs(cutouts)\n",
+        "        if skip_augs is not True: \n",
+        "            for i in range(cutouts.shape[0]):  cutouts[i]=self.augs(cutouts[i])\n",
         "        return cutouts\n",
         "\n",
         "def spherical_dist_loss(x, y):\n",