diff --git a/test/examples/main.cpp b/test/examples/main.cpp
index 50b47cf51..08d9895f0 100644
--- a/test/examples/main.cpp
+++ b/test/examples/main.cpp
@@ -10,6 +10,7 @@
 #include "AllSteps.hpp"
 #include "zklog.hpp"
 #include "exit_process.hpp"
+#include "cuda_utils.hpp"
 
 int main()
 {
@@ -37,6 +38,10 @@ int main()
 
     StarkInfo starkInfo(starkInfoFile, reduceMemory);
 
+#if defined(__USE_CUDA__) && defined(ENABLE_EXPERIMENTAL_CODE)
+    alloc_pinned_mem_per_device((1 << starkInfo.starkStruct.nBitsExt) * 32);
+#endif
+
     uint64_t polBits = starkInfo.starkStruct.steps[starkInfo.starkStruct.steps.size() - 1].nBits;
     FRIProof fproof((1 << polBits), FIELD_EXTENSION, starkInfo.starkStruct.steps.size(), starkInfo.evMap.size(), starkInfo.nPublics);