diff --git a/test/test_data_generator.py b/test/test_data_generator.py
index 97122a265b911524a06372f7e5a99f02fcf7dd3f..ab6233f34533fa7ad35f05306f3990209b83bb82 100644
--- a/test/test_data_generator.py
+++ b/test/test_data_generator.py
@@ -62,7 +62,7 @@ class TestDataGenerator:
         assert station[1].data.shape[-1] == gen.window_lead_time
         assert station[0].data.shape[1] == gen.window_history + 1
 
-    def test_get_key_representation(self, gen):
+    def test_get_station_key(self, gen):
         gen.stations.append("DEBW108")
         f = gen.get_station_key
         iter(gen)