diff --git a/tests/acceptance/test_softmaxAC_mc.py b/tests/acceptance/test_softmaxAC_mc.py
index 9f668d5..9869bd3 100644
--- a/tests/acceptance/test_softmaxAC_mc.py
+++ b/tests/acceptance/test_softmaxAC_mc.py
@@ -1,9 +1,9 @@
 import os
 import pytest
 
-from ml_experiment.experiment_definition import ExperimentDefinition
 from ml_experiment.definition_part import DefinitionPart
-from ml_experiment.Scheduler import LocalRunConfig, RunSpec, Scheduler
+from ml_experiment.experiment_definition import ExperimentDefinition
+from ml_experiment.Scheduler import LocalRunConfig, Scheduler
 
 
 @pytest.fixture