diff --git a/integration_tests/models/16x8/test_concatenate/generate.py b/integration_tests/models/16x8/test_concatenate/generate.py
index f54976c91..ef4300ee9 100644
--- a/integration_tests/models/16x8/test_concatenate/generate.py
+++ b/integration_tests/models/16x8/test_concatenate/generate.py
@@ -6,7 +6,7 @@
 
 def generate_concatenate_model(input_shapes, axis):
     input_data = [tf.keras.Input(shape=input_shape, dtype=np.float32, batch_size=1) for input_shape in input_shapes]
-    concatenated_output = tf.keras.layers.Concatenate(axis=axis)(input_data)
+    concatenated_output = tf.concat(input_data, axis=axis)
     model = tf.keras.Model(inputs=input_data, outputs=concatenated_output)
     converter = tfl.TFLiteConverter.from_keras_model(model)
     def representative_dataset_gen():
@@ -28,6 +28,7 @@ def representative_dataset_gen():
 
 generate_concatenate_model([(64), (64)], 0)
 generate_concatenate_model([(2, 3), (2, 3)], 1)
-generate_concatenate_model([(2, 3, 5), (2, 3, 5)], 0)
+generate_concatenate_model([(2, 3, 5), (2, 3, 5)], -1)
 generate_concatenate_model([(2, 6, 5, 2), (2, 6, 5, 2)], 1)
+generate_concatenate_model([(2, 6, 5, 2), (2, 6, 5, 2)], 3)
 
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite
index 4d3e93b82..321e6ff57 100644
Binary files a/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite and b/integration_tests/models/16x8/test_concatenate/test_concatenate_0.tflite differ
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite
index 7db0ff5a0..603db4a0e 100644
Binary files a/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite and b/integration_tests/models/16x8/test_concatenate/test_concatenate_1.tflite differ
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite
index 1d833a72c..1f00be738 100644
Binary files a/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite and b/integration_tests/models/16x8/test_concatenate/test_concatenate_2.tflite differ
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite
index d25c2568d..77bf16ad7 100644
Binary files a/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite and b/integration_tests/models/16x8/test_concatenate/test_concatenate_3.tflite differ
diff --git a/integration_tests/models/16x8/test_concatenate/test_concatenate_4.tflite b/integration_tests/models/16x8/test_concatenate/test_concatenate_4.tflite
new file mode 100644
index 000000000..eda42c205
Binary files /dev/null and b/integration_tests/models/16x8/test_concatenate/test_concatenate_4.tflite differ