diff --git a/river/stats/test_ks.py b/river/stats/test_ks.py
index 6881e645e0..f3a062b3de 100644
--- a/river/stats/test_ks.py
+++ b/river/stats/test_ks.py
@@ -1,11 +1,12 @@
 from __future__ import annotations
 
 import numpy as np
-import river
 
-from collections import deque
 from scipy.stats import ks_2samp
 
+from collections import deque
+from river import stats
+
 
 def test_incremental_ks_statistics():
     initial_a = np.random.normal(loc=0, scale=1, size=5)
@@ -15,7 +16,7 @@ def test_incremental_ks_statistics():
     stream_b = np.random.normal(loc=1, scale=1, size=10)
 
     incremental_ks_statistics = []
-    incremental_ks = river.stats.KolmogorovSmirnov(statistic="ks")
+    incremental_ks = stats.KolmogorovSmirnov(statistic="ks")
     sliding_a = deque(initial_a)
     sliding_b = deque(initial_b)