diff --git a/.github/workflows/quality-check.yaml b/.github/workflows/quality-check.yaml
index 4f62c303..765aa9ca 100644
--- a/.github/workflows/quality-check.yaml
+++ b/.github/workflows/quality-check.yaml
@@ -58,7 +58,6 @@ jobs:
               keras: "~=2.15.0"
               tensorflow-probability: "~=0.23.0"
             - tensorflow: "~=2.16.0"
-              keras: "~=2.16.0"
               tensorflow-probability: "~=0.24.0"
           exclude:
             # These older versions of TensorFlow don't work with Python 3.10:
diff --git a/tests/integration/test_compilation.py b/tests/integration/test_compilation.py
index c1bd02d6..12b1ea9f 100644
--- a/tests/integration/test_compilation.py
+++ b/tests/integration/test_compilation.py
@@ -16,7 +16,6 @@
 import numpy as np
 import pytest
 import tensorflow as tf
-from tensorflow import keras
 
 from gpflow.keras import tf_keras
 from gpflow.kernels import RBF
@@ -66,7 +65,7 @@ def build_keras_functional_deep_gp(layer_sizes, num_data):
     gp_layers = build_gp_layers(layer_sizes, num_data)
     likelihood = Gaussian()
 
-    inputs = keras.Input(shape=(layer_sizes[0]))
+    inputs = tf_keras.Input(shape=(layer_sizes[0]))
     x = inputs
     for gp in gp_layers:
         x = gp(x)