diff --git a/tests/test_progit_flask_ui_app.py b/tests/test_progit_flask_ui_app.py
index feceee0..7b1624a 100644
--- a/tests/test_progit_flask_ui_app.py
+++ b/tests/test_progit_flask_ui_app.py
@@ -39,6 +39,7 @@ class ProgitFlaskApptests(tests.Modeltests):
         progit.APP.config['TESTING'] = True
         progit.SESSION = self.session
         progit.ui.SESSION = self.session
+        progit.ui.app.SESSION = self.session
         self.app = progit.APP.test_client()
         progit.APP.config['GIT_FOLDER'] = HERE
         progit.APP.config['FORK_FOLDER'] = HERE