diff --git a/cca_zoo/models/__init__.py b/cca_zoo/models/__init__.py
index fa236562..286880ba 100644
--- a/cca_zoo/models/__init__.py
+++ b/cca_zoo/models/__init__.py
@@ -15,11 +15,6 @@
 from ._rcca import rCCA, CCA, PLS
 from ._tcca import TCCA, KTCCA
 
-try:
-    from ._stochastic import StochasticPowerPLS, IncrementalPLS
-except:
-    pass
-
 __all__ = [
     "GCCA",
     "KGCCA",
@@ -40,8 +35,14 @@
     "PLS",
     "TCCA",
     "KTCCA",
-    "StochasticPowerPLS",
-    "IncrementalPLS",
 ]
 
+try:
+    from ._stochastic import StochasticPowerPLS, IncrementalPLS
+
+    __all__.extend("StochasticPowerPLS",
+                   "IncrementalPLS")
+except:
+    pass
+
 classes = __all__