diff --git a/tests/cli/13_test_stack.py b/tests/cli/13_test_stack.py
index 63939cb..c6b19b3 100644
--- a/tests/cli/13_test_stack.py
+++ b/tests/cli/13_test_stack.py
@@ -10,10 +10,11 @@ class CliTestCaseStack(test.WOTestCase):
         self.app.close()
 
     def test_wo_cli_stack_install(self):
-        self.app = get_test_app(argv=['stack', 'install'])
-        self.app.setup()
-        self.app.run()
-        self.app.close()
+        argv = ['stack', 'install']
+        with self.make_app(argv=argv) as app:
+            app.run()
+            self.eq(app.pargs.stack.install)
+
 
     def test_wo_cli_stack_install_web(self):
         self.app = get_test_app(argv=['stack', 'install', '--web'])
diff --git a/tests/cli/plugins/test_example.py b/tests/cli/plugins/test_example.py
index 5de3c2d..dc101f8 100644
--- a/tests/cli/plugins/test_example.py
+++ b/tests/cli/plugins/test_example.py
@@ -1,6 +1,6 @@
 """Tests for Example Plugin."""
 
-from wo.utils import WOTestCase
+from wo.utils.test import WOTestCase
 
 
 class ExamplePluginTestCase(WOTestCase):