diff --git a/lama/saicinpainting/training/modules/fake_fakes.py b/lama/saicinpainting/training/modules/fake_fakes.py
index 45c4ad55..eff44836 100644
--- a/lama/saicinpainting/training/modules/fake_fakes.py
+++ b/lama/saicinpainting/training/modules/fake_fakes.py
@@ -1,5 +1,5 @@
 import torch
-from kornia import SamplePadding
+from kornia.constants import SamplePadding
 from kornia.augmentation import RandomAffine, CenterCrop