1 files changed, 2 insertions, 1 deletions
diff --git a/tests/datasets/ActivationFunctionsDataset.h b/tests/datasets/ActivationFunctionsDataset.h
index 147c5ae51b..c5dc28f0c3 100644
--- a/tests/datasets/ActivationFunctionsDataset.h
+++ b/tests/datasets/ActivationFunctionsDataset.h
@@ -51,7 +51,8 @@ public:
                             ActivationLayerInfo::ActivationFunction::SOFT_RELU,
                             ActivationLayerInfo::ActivationFunction::SQRT,
                             ActivationLayerInfo::ActivationFunction::SQUARE,
-                            ActivationLayerInfo::ActivationFunction::TANH
+                            ActivationLayerInfo::ActivationFunction::TANH,
+                            ActivationLayerInfo::ActivationFunction::IDENTITY
     })
     {
     }