diff --git a/tests/test_random_generator.py b/tests/test_random_generator.py
index 5eb8395..c5657da 100644
--- a/tests/test_random_generator.py
+++ b/tests/test_random_generator.py
@@ -1,7 +1,7 @@
 import matplotlib.pyplot as plt
 import pytest
 
-from cluster_import import from_csv
+from cluster_import import import_thermal_cluster
 from ts_generator import ProbilityLaw, ThermalCluster, ThermalDataGenerator