diff --git a/gpax/__init__.py b/gpax/__init__.py
index 9a1df20..7529e8f 100644
--- a/gpax/__init__.py
+++ b/gpax/__init__.py
@@ -4,7 +4,7 @@
 from . import acquisition
 from .hypo import sample_next
 from .models import (DKL, CoregGP, ExactGP, MultiTaskGP, iBNN, vExactGP,
-                     vi_iBNN, viDKL, viGP, viMTDKL, VarNoiseGP, UIGP,
+                     vi_iBNN, viDKL, viGP, sPM, viMTDKL, VarNoiseGP, UIGP,
                      MeasuredNoiseGP, viSparseGP)
 
 __all__ = ["utils", "kernels", "mtkernels", "acquisition", "ExactGP", "vExactGP", "DKL",