diff --git a/river/proba/__init__.py b/river/proba/__init__.py
index 49536c626e..c3e399534a 100644
--- a/river/proba/__init__.py
+++ b/river/proba/__init__.py
@@ -3,8 +3,7 @@
 
 from . import base
 from .beta import Beta
-from .gaussian import Gaussian
-from .gaussian import MultivariateGaussian
+from .gaussian import Gaussian, MultivariateGaussian
 from .multinomial import Multinomial
 
 __all__ = ["base", "Beta", "Gaussian", "Multinomial", "MultivariateGaussian"]
diff --git a/river/proba/gaussian.py b/river/proba/gaussian.py
index 86e37a596c..2c8f51ae28 100644
--- a/river/proba/gaussian.py
+++ b/river/proba/gaussian.py
@@ -1,15 +1,13 @@
 from __future__ import annotations
 
 import math
-import pytest
-import warnings
 
 import numpy as np
 import pandas as pd
+import pytest
 from scipy.stats import multivariate_normal
 
-from river import covariance
-from river import stats
+from river import covariance, stats
 from river.proba import base
 
 __all__ = ["Gaussian", "MultivariateGaussian"]