diff --git a/pyproject.toml b/pyproject.toml
index 0ae9b809..6388767f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -12,7 +12,7 @@ exclude = ["build*"]
 "simopt.data_farming" = ["*.npz"]
 
 [tool.uv.sources]
-simopt-extensions = { git = "https://github.com/cenwangumass/simopt-extensions", rev = "2bcf002" }
+simopt-extensions = { git = "https://github.com/cenwangumass/simopt-extensions", rev = "b40a9c8" }
 
 [project]
 name = "simoptlib"