diff --git a/test/Convolution2D.hpp b/test/Convolution2D.hpp
index 46b4654..180f57e 100644
--- a/test/Convolution2D.hpp
+++ b/test/Convolution2D.hpp
@@ -102,7 +102,7 @@
     AddPoolAndSetData(6, request, indata);
 
     // add memory for the output
-    android::sp<IMemory> outMemory = AddPoolAndGetData(outSize, request);
+    android::sp<IMemory> outMemory = AddPoolAndGetData<float>(outSize, request);
     float* outdata = reinterpret_cast<float*>(static_cast<void*>(outMemory->getPointer()));
 
     // run the execution