diff --git a/olt.py b/olt.py
index 4b2897d..82b520f 100644
--- a/olt.py
+++ b/olt.py
@@ -770,6 +770,7 @@
 
     def runTest(self):
         logging.info("Testing Rule removal")
+	delete_all_flows(self.controller)
         processEapolRule(self, onu_port)
         #wait for the rule to settle
         time.sleep(3)
@@ -780,7 +781,7 @@
 
         stats = get_flow_stats(self, ofp.match())
 
-        test.assertTrue(len(stats) == 3, \
+        self.assertTrue(len(stats) == 3, \
                         "Wrong number of rules reports; reported %s, expected 3\n\n %s" % (len(stats), stats))
 
         processEapolRule(self, onu_port, install = False)
@@ -788,7 +789,7 @@
 
         stats = get_flow_stats(self, ofp.match())
 
-        test.assertTrue(len(stats) == 2, \
+        self.assertTrue(len(stats) == 2, \
                         "Wrong number of rules reports; reported %s, expected 2\n\n %s" % (len(stats), stats))
 
         logging.info(stats)