1 files changed, 6 insertions, 2 deletions
diff --git a/fabfile.py b/fabfile.py
index 007fd7f..472c769 100644
--- a/fabfile.py
+++ b/fabfile.py
@@ -63,17 +63,21 @@ def build_tests(version=_pyversion[:3]):
 
 def generate_test(file):
     """ Generate a given test. """
+    import sys, os
+    sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
     import tests
     config = tests.get_config(os.path.dirname(file))
     root, ext = os.path.splitext(file)
-    if ext == config.get(get_section(os.path.basename(root), config), 
+    if ext == config.get(tests.get_section(os.path.basename(root), config), 
                          'input_ext'):
         tests.generate(root, config)
     else:
-        print test, 'does not have a valid file extension. Check config.'
+        print file, 'does not have a valid file extension. Check config.'
 
 def generate_tests():
     """ Generate all outdated tests. """
+    import sys, os
+    sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
     from tests import generate_all
     generate_all()