diff --git a/graph_framework/cuda_context.hpp b/graph_framework/cuda_context.hpp
index c1d2cbb89067e8039a69b54cc596bff01610bb2c..e3c9ce0c6085c927599f4732ddc0f879cd083c7e 100644
--- a/graph_framework/cuda_context.hpp
+++ b/graph_framework/cuda_context.hpp
@@ -723,7 +723,7 @@ namespace gpu {
                 jit::add_type<T> (source_buffer);
                 source_buffer << " * __restrict__ "
                               << jit::to_string('v', inputs[0].get());
-                sed_args.insert(inputs[0].get());
+                used_args.insert(inputs[0].get());
             }
             for (size_t i = 1, ie = inputs.size(); i < ie; i++) {
                 if (!used_args.contains(inputs[i].get())) {