diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py
index fbb2a644ec170b3695de59499876e1b0e420fe08..6dee37e626cc9899dd0a3a5f90cd16aef2620b7f 100644
--- a/src/pystencils_autodiff/framework_integration/astnodes.py
+++ b/src/pystencils_autodiff/framework_integration/astnodes.py
@@ -102,7 +102,7 @@ class FunctionCall(Node):
 
     @property
     def args(self):
-        return [p.symbol for p in self.kernel_function.get_parameters()]
+        return [p.symbol for p in self.kernel_function.get_parameters()] + [self.kernel_function]
 
     @property
     def symbols_defined(self) -> Set[sp.Symbol]: