diff --git a/integration_tests/dataset_tests/imdb_test.py b/integration_tests/dataset_tests/imdb_test.py
index e2971c470..a41bf6f97 100644
--- a/integration_tests/dataset_tests/imdb_test.py
+++ b/integration_tests/dataset_tests/imdb_test.py
@@ -7,9 +7,9 @@ from keras.src.datasets import imdb
 class ImdbLoadDataTest(testing.TestCase):
     def test_load_data_default(self):
         (x_train, y_train), (x_test, y_test) = imdb.load_data()
-        self.assertIsInstance(x_train, list)
+        self.assertIsInstance(x_train, np.ndarray)
         self.assertIsInstance(y_train, np.ndarray)
-        self.assertIsInstance(x_test, list)
+        self.assertIsInstance(x_test, np.ndarray)
         self.assertIsInstance(y_test, np.ndarray)
 
         # Check lengths