diff --git a/setup.cfg b/setup.cfg
index 982ea666d..73e6ffdc6 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -30,7 +30,7 @@ install_requires =
     fastcore >=1.3.0,<1.4
     tqdm >=4.49.0,<5
     opencv-python >=4.1.1,<5
-    albumentations >=0.4.6,<0.6
+    albumentations >=1.0.3,<1.1
     matplotlib >=3.2.2,<4
     pycocotools>=2.0.2,<3
     requests >=2.23.0,<3
diff --git a/tests/transforms/test_albu_transform.py b/tests/transforms/test_albu_transform.py
index bfb2f557a..0c85df040 100644
--- a/tests/transforms/test_albu_transform.py
+++ b/tests/transforms/test_albu_transform.py
@@ -3,7 +3,7 @@
 )
 import pytest
 from icevision.all import *
-from albumentations.augmentations.transforms import (
+from albumentations import (
     LongestMaxSize,
     Normalize,
     PadIfNeeded,