diff --git a/python-package/setup.py b/python-package/setup.py
index 3d73150e1..a48eb811c 100644
--- a/python-package/setup.py
+++ b/python-package/setup.py
@@ -54,7 +54,7 @@ def find_version(*file_paths):
     'scikit-image',
     'easydict',
     'cython',
-    'albumentations',
+    'albumentationsx',
     'prettytable',
 ]