diff --git a/tests/framework/instruments/OpenCLMemoryUsage.cpp b/tests/framework/instruments/OpenCLMemoryUsage.cpp
index b090bf1..119ad4c 100644
--- a/tests/framework/instruments/OpenCLMemoryUsage.cpp
+++ b/tests/framework/instruments/OpenCLMemoryUsage.cpp
@@ -124,6 +124,7 @@
             void *retval = this->real_clSVMAlloc(context, flags, size, alignment);
             if(retval != nullptr)
             {
+                _svm_allocations[retval] = size;
                 _now.num_allocations++;
                 _now.in_use += size;
                 _now.total_allocated += size;