test code working
diff --git a/wally/main.py b/wally/main.py
index f57e1a5..6396b6b 100644
--- a/wally/main.py
+++ b/wally/main.py
@@ -40,7 +40,8 @@
from .ceph import DiscoverCephStage
from .openstack import DiscoverOSStage
from .fuel import DiscoverFuelStage
-from .run_test import CollectInfoStage, ExplicitNodesStage, SaveNodesStage, RunTestsStage, ConnectStage, SleepStage
+from .run_test import (CollectInfoStage, ExplicitNodesStage, SaveNodesStage,
+ RunTestsStage, ConnectStage, SleepStage, PrepareNodes)
from .report import ConsoleReportStage, HtmlReportStage
from .sensors import StartSensorsStage, CollectSensorsStage
@@ -219,6 +220,7 @@
stages.append(CollectSensorsStage())
stages.append(ConnectStage())
stages.append(SleepStage())
+ stages.append(PrepareNodes())
if not opts.dont_collect:
stages.append(CollectInfoStage())