diff --git a/tests/testapp/application.py b/tests/testapp/application.py
index 116efc66..af5f5b33 100644
--- a/tests/testapp/application.py
+++ b/tests/testapp/application.py
@@ -17,7 +17,8 @@
 from os import environ
 
 import user_profile_service
-from flask import CSRFProtect, Flask, request
+from flask import Flask, request
+from flask_wtf.csrf import CSRFProtect
 
 from optimizely import logger, optimizely
 from optimizely.helpers import enums
diff --git a/tests/testapp/requirements.txt b/tests/testapp/requirements.txt
index 4b70123b..dae26c1f 100644
--- a/tests/testapp/requirements.txt
+++ b/tests/testapp/requirements.txt
@@ -1 +1,2 @@
-Flask==2.2.5
+Flask==3.1.0
+flask-wtf==1.2.2
\ No newline at end of file