diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
index e20e900..6b8e403 100644
--- a/lightweight_gan/lightweight_gan.py
+++ b/lightweight_gan/lightweight_gan.py
@@ -22,7 +22,7 @@
 from PIL import Image
 import torchvision
 from torchvision import transforms
-from kornia import filter2D
+from kornia import filter2d
 
 from lightweight_gan.diff_augment import DiffAugment
 from lightweight_gan.version import __version__
@@ -190,7 +190,7 @@ def __init__(self):
     def forward(self, x):
         f = self.f
         f = f[None, None, :] * f [None, :, None]
-        return filter2D(x, f, normalized=True)
+        return filter2d(x, f, normalized=True)
 
 # attention
 
diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
index abadaef..dfe5310 100644
--- a/lightweight_gan/version.py
+++ b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.20.1'
+__version__ = '0.20.2'
diff --git a/setup.py b/setup.py
index 4824f32..e69ee4b 100644
--- a/setup.py
+++ b/setup.py
@@ -27,7 +27,7 @@
     'adabelief-pytorch',
     'einops>=0.3',
     'fire',
-    'kornia',
+    'kornia>=0.5.4',
     'numpy',
     'pillow',
     'retry',