diff --git a/gpax/acquisition/__init__.py b/gpax/acquisition/__init__.py
index aee6c5d..08ebdf1 100644
--- a/gpax/acquisition/__init__.py
+++ b/gpax/acquisition/__init__.py
@@ -1,4 +1,5 @@
 from .acquisition import UCB, EI, POI, UE, Thompson, KG
 from .batch_acquisition import qEI, qPOI, qUCB, qKG
+from .optimize import optimize_acq
 
-__all__ = ["UCB", "EI", "POI", "UE", "KG", "Thompson", "qEI", "qPOI", "qUCB", "qKG"]
+__all__ = ["UCB", "EI", "POI", "UE", "KG", "Thompson", "qEI", "qPOI", "qUCB", "qKG", "optimize_acq"]