diff --git a/server/pyproject.toml b/server/pyproject.toml
index 1cf45996..b468713b 100644
--- a/server/pyproject.toml
+++ b/server/pyproject.toml
@@ -33,7 +33,7 @@ onnxruntime-gpu = { version = "1.18.0", optional = true }
 onnx = { version = "^1.16.0", optional = true }
 einops = "^0.7.0"
 ibm-fms = { version = "^0.0", optional = true }
-fms-extras = { git = "https://github.com/foundation-model-stack/fms-extras", rev = "d41f8a34c9841aa3c4c59f17b5e7f3cb365f49de", optional = true }
+fms-extras = { git = "https://github.com/foundation-model-stack/fms-extras", rev = "16339f7c82255983d20dabf4807d124133be1d1c", optional = true }
 
 # Hold back breaking changes from numpy 2.0
 numpy = "<2.0"