diff --git a/test/test_model_modules/test_flatten_tail.py b/test/test_model_modules/test_flatten_tail.py
index 8a858b438357ff3b39834fb0aea01967df8f2f9c..b53e381ea1cecc1d6dfbe019264c726f11946479 100644
--- a/test/test_model_modules/test_flatten_tail.py
+++ b/test/test_model_modules/test_flatten_tail.py
@@ -62,7 +62,7 @@ class TestFlattenTail:
         assert final_dense.units == 2
         assert final_dense.kernel_regularizer is None
         inner_act = self.step_in(final_dense)
-        assert inner_act.name == 'tf.nn.elu'
+        assert inner_act.name == 'tf.nn.elu_1'
         assert inner_act.__class__.__name__ == 'TFOpLambda'
         inner_dense = self.step_in(inner_act)
         assert inner_dense.name == 'Main_tail_inner_Dense'