diff --git a/Dockerfile b/Dockerfile
index 3d94fc4b..ad4cfd33 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -479,6 +479,7 @@ RUN pip install flashtext && \
     pip install optuna && \
     pip install chainercv && \
     pip install plotly_express && \
+    pip install albumentations && \
     /tmp/clean-layer.sh
 
 # Tesseract and some associated utility packages
diff --git a/tests/test_albumentations.py b/tests/test_albumentations.py
new file mode 100644
index 00000000..677c90c2
--- /dev/null
+++ b/tests/test_albumentations.py
@@ -0,0 +1,10 @@
+import unittest
+
+import cv2
+from albumentations import HorizontalFlip
+
+class TestAlbumentations(unittest.TestCase):
+    def test_rotate(self):
+        image = cv2.imread('/input/tests/data/dot.png')
+        aug = HorizontalFlip(p=1)
+        image_rotated = aug(image=image)['image']