X-Git-Url: https://scripts.mit.edu/gitweb/wizard.git/blobdiff_plain/3f56e5237e400234761729ddae0d0b2a9ca74085..6554c6378a6e801b4fe47c50688cdae1d627bc18:/wizard/tests/deploy_test.py

diff --git a/wizard/tests/deploy_test.py b/wizard/tests/deploy_test.py
index 95e7bb8..9de6bcb 100644
--- a/wizard/tests/deploy_test.py
+++ b/wizard/tests/deploy_test.py
@@ -2,7 +2,7 @@ import distutils.version
 import datetime
 import dateutil.tz
 
-from wizard import deploy
+from wizard import app, deploy
 
 def test_deployment_parse():
     result = deploy.Deployment.parse("/afs/athena.mit.edu/user/e/z/ezyang/web_scripts/test-wiki:/afs/athena.mit.edu/contrib/scripts/deploy/mediawiki-1.11.0\n")
@@ -14,7 +14,7 @@ def test_deployment_parse_nosuchapplication():
     try:
         deploy.Deployment.parse("a:/foo/obviouslybogus-1.11.0\n")
         assert False
-    except deploy.NoSuchApplication:
+    except app.NoSuchApplication:
         pass
 
 def test_deployment_from_dir():