diff --git a/tests/test_pipreqs.py b/tests/test_pipreqs.py
index c7c65eb..b24a8b8 100755
--- a/tests/test_pipreqs.py
+++ b/tests/test_pipreqs.py
@@ -107,8 +107,7 @@ class TestPipreqs(unittest.TestCase):
 
 
     def test_custom_pypi_server(self):
-        with self.assertRaises(requests.exceptions.MissingSchema):
-            pipreqs.init({'<path>': self.project, '--savepath': None,
+        self.assertRaises(requests.exceptions.MissingSchema, pipreqs.init, {'<path>': self.project, '--savepath': None,
                       '--use-local': None, '--force': True, '--proxy':None, '--pypi-server':'nonexistent'})