diff --git "a/notebooks/\316\274NCA_pytorch.ipynb" "b/notebooks/\316\274NCA_pytorch.ipynb"
index 46fd5f5..f5b3ef5 100644
--- "a/notebooks/\316\274NCA_pytorch.ipynb"
+++ "b/notebooks/\316\274NCA_pytorch.ipynb"
@@ -270,7 +270,7 @@
       "source": [
         "# OT loss \n",
         "from geomloss import SamplesLoss\n",
-        "from torchvision.transforms.functional_tensor import gaussian_blur\n",
+        "from torchvision.transforms.functional import gaussian_blur\n",
         "\n",
         "def calc_styles_ot(img, ksize, leveln, show=False):\n",
         "  levels = []\n",
@@ -784,4 +784,4 @@
       ]
     }
   ]
-}
\ No newline at end of file
+}