2 files changed, 6 insertions, 2 deletions

diff --git a/tests/test_error.py b/tests/test_error.py
index 2733ab8c..2b28f5c2 100644
--- a/tests/test_error.py
+++ b/tests/test_error.py
@@ -21,12 +21,16 @@ import unittest
 import error
 import project
 import git_command
+import fetch
+import command
 from subcmds import all_modules
 
 imports = all_modules + [
     error,
     project,
     git_command,
+    fetch,
+    command,
 ]
 
 
diff --git a/tests/test_git_command.py b/tests/test_git_command.py
index 1e8beabc..3dd31b29 100644
--- a/tests/test_git_command.py
+++ b/tests/test_git_command.py
@@ -204,12 +204,12 @@ class GitRequireTests(unittest.TestCase):
 
     def test_older_fatal(self):
         """Test fatal require calls with old versions."""
-        with self.assertRaises(SystemExit) as e:
+        with self.assertRaises(git_command.GitRequireError) as e:
             git_command.git_require((2,), fail=True)
             self.assertNotEqual(0, e.code)
 
     def test_older_fatal_msg(self):
         """Test fatal require calls with old versions and message."""
-        with self.assertRaises(SystemExit) as e:
+        with self.assertRaises(git_command.GitRequireError) as e:
             git_command.git_require((2,), fail=True, msg="so sad")
             self.assertNotEqual(0, e.code)