diff --git a/graph_framework/cuda_context.hpp b/graph_framework/cuda_context.hpp
index f91090e887d439645440ae821ae9cf3ffa67bdbb..8dfa9f438ca7edbe5b1d0fd5f2b8b063ae0d14c5 100644
--- a/graph_framework/cuda_context.hpp
+++ b/graph_framework/cuda_context.hpp
@@ -343,7 +343,7 @@ namespace gpu {
                     needed_buffers.insert(input.get());
                 }
                 if (!needed_buffers.contains(input.get())) {
-                    buffers.push_back(kernel_arguments[input.get()]);
+                    buffers.push_back(reinterpret_cast<void *> (&kernel_arguments[input.get()]));
                     needed_buffers.insert(input.get());
                 }
             }
@@ -358,7 +358,7 @@ namespace gpu {
                     needed_buffers.insert(output.get());
                 }
                 if (!needed_buffers.contains(output.get())) {
-                    buffers.push_back(kernel_arguments[output.get()]);
+                    buffers.push_back(reinterpret_cast<void *> (&kernel_arguments[input.get()]));
                     needed_buffers.insert(output.get());
                 }
             }