diff --git a/pysindy/__init__.py b/pysindy/__init__.py
index c3b4e71a..b7f929a1 100644
--- a/pysindy/__init__.py
+++ b/pysindy/__init__.py
@@ -36,6 +36,7 @@
 from .optimizers import SSR
 from .optimizers import STLSQ
 from .optimizers import EnsembleOptimizer
+from .optimizers import SINDyPI
 
 try:
     from .optimizers import ConstrainedSR3
diff --git a/pysindy/optimizers/__init__.py b/pysindy/optimizers/__init__.py
index 4ef8f36d..67237431 100644
--- a/pysindy/optimizers/__init__.py
+++ b/pysindy/optimizers/__init__.py
@@ -4,6 +4,7 @@
 from .sr3 import SR3
 from .ssr import SSR
 from .stlsq import STLSQ
+from .sindy_pi import SINDyPI
 from .wrapped_optimizer import WrappedOptimizer
 
 try: