diff --git a/tests/smoke/integrate/test_package.py b/tests/smoke/integrate/test_package.py
index 216e5323..6710528a 100644
--- a/tests/smoke/integrate/test_package.py
+++ b/tests/smoke/integrate/test_package.py
@@ -1,9 +1,11 @@
 """Test the :mod:`galax.integrate` module."""
 
 from galax import integrate
-from galax.integrate import _base, _builtin
+from galax.integrate import _api, _base, _builtin
 
 
 def test_all() -> None:
     """Test the API."""
-    assert set(integrate.__all__) == set(_base.__all__ + _builtin.__all__)
+    assert set(integrate.__all__) == set(
+        _api.__all__ + _base.__all__ + _builtin.__all__
+    )