diff --git a/pyproject.toml b/pyproject.toml
index d85a121a2..bf02155ae 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -66,7 +66,7 @@ cloudvol = [
 ]
 augmentations = [
     "zetta_utils[tensor_ops]",
-    "imgaug == 0.4.0",
+    "imgaug @ git+https://github.com/u7122029/imgaug.git@418f22d4275e9f90274900e27b595ed678bb4bcc",
     "imagecorruptions == 1.1.2",
 ]
 convnet = ["torch >= 2.0", "artificery >= 0.0.3.3"]