diff --git a/tests/test_progit_flask_internal.py b/tests/test_progit_flask_internal.py
index 9931a85..7300ceb 100644
--- a/tests/test_progit_flask_internal.py
+++ b/tests/test_progit_flask_internal.py
@@ -39,6 +39,8 @@ class PagureFlaskInternaltests(tests.Modeltests):
             pagure.APP.config['IP_ALLOWED_INTERNAL'] + [None]))
         pagure.SESSION = self.session
         pagure.internal.SESSION = self.session
+        pagure.ui.repo.SESSION = self.session
+
         pagure.APP.config['GIT_FOLDER'] = tests.HERE
         pagure.APP.config['FORK_FOLDER'] = os.path.join(
             tests.HERE, 'forks')
diff --git a/tests/test_progit_flask_ui_fork.py b/tests/test_progit_flask_ui_fork.py
index 66abfbc..2636e17 100644
--- a/tests/test_progit_flask_ui_fork.py
+++ b/tests/test_progit_flask_ui_fork.py
@@ -40,6 +40,7 @@ class PagureFlaskForktests(tests.Modeltests):
         pagure.lib.SESSION = self.session
         pagure.ui.SESSION = self.session
         pagure.ui.app.SESSION = self.session
+        pagure.ui.filters.SESSION = self.session
         pagure.ui.fork.SESSION = self.session
         pagure.ui.repo.SESSION = self.session