diff --git a/mlpp_lib/utils.py b/mlpp_lib/utils.py
index 5a21cc8..6ef69ce 100644
--- a/mlpp_lib/utils.py
+++ b/mlpp_lib/utils.py
@@ -5,7 +5,7 @@
 import xarray as xr
 import tensorflow as tf
 
-from mlpp_lib import callbacks, losses, metrics, models, probabilistic_layers
+from mlpp_lib import callbacks, losses, metrics, models
 
 
 LOGGER = logging.getLogger(__name__)