diff --git a/src/cryojax/inference/distributions/__init__.py b/src/cryojax/inference/distributions/__init__.py
index e888e270..893826d3 100644
--- a/src/cryojax/inference/distributions/__init__.py
+++ b/src/cryojax/inference/distributions/__init__.py
@@ -3,6 +3,7 @@
     AbstractMarginalDistribution as AbstractMarginalDistribution,
 )
 from ._gaussian_distributions import (
+    AbstractGaussianDistribution as AbstractGaussianDistribution,
     IndependentGaussianFourierModes as IndependentGaussianFourierModes,
     IndependentGaussianPixels as IndependentGaussianPixels,
 )