diff --git a/test/Convolution2D.hpp b/test/Convolution2D.hpp
index ec43ae3..46b4654 100644
--- a/test/Convolution2D.hpp
+++ b/test/Convolution2D.hpp
@@ -32,7 +32,7 @@
 
 void SetModelFp16Flag(V1_0::Model& model, bool fp16Enabled);
 
-#ifdef ARMNN_ANDROID_NN_V1_1
+#if defined(ARMNN_ANDROID_NN_V1_1) || defined(ARMNN_ANDROID_NN_V1_2)
 void SetModelFp16Flag(V1_1::Model& model, bool fp16Enabled);
 #endif