diff --git a/services/flask/requirements.txt b/services/flask/requirements.txt
index 9aa8695..97d0ae5 100644
--- a/services/flask/requirements.txt
+++ b/services/flask/requirements.txt
@@ -10,5 +10,4 @@ pytest~=8.2.0
 requests~=2.31.0
 setuptools>=65.5.1 # not directly required, pinned to avoid a vulnerability
 # tensorflow is a testing requirement
-tensorflow~=2.16.1
-tf-keras~=2.16
+tensorflow~=2.15.1
diff --git a/tests/test_model.py b/tests/test_model.py
index 8c20d55..18d323b 100644
--- a/tests/test_model.py
+++ b/tests/test_model.py
@@ -4,7 +4,7 @@
 import cv2
 import numpy as np
 import pytest
-import tf_keras as keras
+import tensorflow as tf
 from conftest import *
 
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"  # Disable tensorflow debugging logs
@@ -12,7 +12,7 @@
 
 @pytest.fixture(scope="module")
 def model():
-    return keras.models.load_model(
+    return tf.keras.models.load_model(
         Path(__file__).parents[1].resolve()
         / "services"
         / "tensorflow-serving"