diff --git a/mininet/test/test_switchdpidassignment.py b/mininet/test/test_switchdpidassignment.py
index e401c605676707e8b7ac2b0c36087237fbfdf4d5..61833e94cb2d683acfe889acac87a1e2d88b1d2a 100644
--- a/mininet/test/test_switchdpidassignment.py
+++ b/mininet/test/test_switchdpidassignment.py
@@ -4,6 +4,7 @@
    Regression tests for switch dpid assignment."""
 
 import unittest
+import sys
 from functools import partial
 import re
 
@@ -13,7 +14,7 @@
 from mininet.topo import Topo
 from mininet.log import setLogLevel
 from mininet.util import quietRun
-
+from mininet.clean import cleanup
 
 class testSwitchDpidAssignmentCommon ( object ):
     """Verify Switch dpid assignment."""