diff --git a/src/pairs/analysis/devices.py b/src/pairs/analysis/devices.py
index 4558953d88131fdc0905eda2870646959aaad33c..0e6126988cb180edbe10200575be239a55be9d55 100644
--- a/src/pairs/analysis/devices.py
+++ b/src/pairs/analysis/devices.py
@@ -48,7 +48,7 @@ class FetchKernelReferences(Visitor):
     def visit_Decl(self, ast_node):
         if isinstance(ast_node.elem, BinOp):
             for k in self.kernel_stack:
-                self.kernel_decls[k.kernel_id].append(ast_node)
+                self.kernel_decls[k.kernel_id].append(ast_node.elem)
 
     def visit_BinOp(self, ast_node):
         for k in self.kernel_stack: