diff --git a/test/unit_tests/conftest.py b/test/unit_tests/conftest.py
index 634d175..c119fa7 100644
--- a/test/unit_tests/conftest.py
+++ b/test/unit_tests/conftest.py
@@ -58,7 +58,7 @@ def keras_model(input_size):
 
     inputs = tf.keras.Input(shape=input_size)
     conv = tf.keras.layers.Conv2D(3, 3)
-    dense = tf.keras.layers.Dense(10, activation="relu")
+    dense = tf.keras.layers.Dense(10)
 
     outputs = tf.keras.layers.Dense(10)(dense(conv(inputs)))
     model = tf.keras.Model(inputs=inputs, outputs=outputs)