diff --git a/gpax/models/__init__.py b/gpax/models/__init__.py
index 84f2f9a..29e2d89 100644
--- a/gpax/models/__init__.py
+++ b/gpax/models/__init__.py
@@ -1,6 +1,7 @@
 from .gp import ExactGP
 from .vgp import vExactGP
 from .vigp import viGP
+from hskgp import VarNoiseGP
 from .spm import sPM
 from .ibnn import iBNN
 from .vi_ibnn import vi_iBNN