diff --git a/test/test_feedback.py b/test/test_feedback.py
index c37d8481..4f119fbf 100644
--- a/test/test_feedback.py
+++ b/test/test_feedback.py
@@ -1,7 +1,6 @@
 import unittest
 import subprocess
 from path import path
-import __builtin__
 
 import autowig
 
@@ -106,7 +105,7 @@ def test_with_none_overload_export(self, overload="none"):
                                   silent = True)
 
         autowig.controller.plugin = 'default'
-        autowig.controller(asg)
+        autowig.controller(asg, overload=overload)
 
         module = autowig.generator(asg, module = self.srcdir/'_module.cpp',
                                      decorator = self.srcdir/'_module.py',
@@ -118,7 +117,8 @@ def test_with_none_overload_export(self, overload="none"):
         module.write()
 
 
-        s = subprocess.Popen(['scons', 'py'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        s = subprocess.Popen(['scons', 'py'],
+                             stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         prev, curr = s.communicate()
 
         while not prev == curr:
@@ -137,4 +137,4 @@ def test_with_class_overload_export(self):
 
     def test_with_namespace_overload_export(self):
         """Test `feedback` with 'namespace' overload"""
-        self.test_with_none_overload_export(overload="namespace")
\ No newline at end of file
+        self.test_with_none_overload_export(overload="namespace")