diff --git a/src/pathcomp/frontend/tests/test_unitary_pathcomp_forecaster.py b/src/pathcomp/frontend/tests/test_unitary_pathcomp_forecaster.py
index 31fe08cd6aadc339114723a833bce91adc4780dd..d6d39d3177f1a3d2275a6b48d68478c0a37e9d6a 100644
--- a/src/pathcomp/frontend/tests/test_unitary_pathcomp_forecaster.py
+++ b/src/pathcomp/frontend/tests/test_unitary_pathcomp_forecaster.py
@@ -75,9 +75,7 @@ JSON_ADMIN_CONTEXT_ID = json_context_id(DEFAULT_CONTEXT_NAME)
 ADMIN_CONTEXT_ID = ContextId(**JSON_ADMIN_CONTEXT_ID)
 ADMIN_TOPOLOGY_ID = TopologyId(**json_topology_id(DEFAULT_TOPOLOGY_NAME, context_id=JSON_ADMIN_CONTEXT_ID))
 
-CSV_DATA_FILE = 'forecaster/tests/data/dataset-short.csv'
-#CSV_DATA_FILE = 'forecaster/tests/data/dataset.csv'
-#CSV_DATA_FILE = 'forecaster/tests/data/dataset2.csv'
+CSV_DATA_FILE = 'forecaster/tests/data/dataset.csv'
 #DESC_DATS_FILE = 'forecaster/tests/data/descriptor.json'
 
 @pytest.fixture(scope='session')