diff --git a/examples/cfd/aero_graph_net/models.py b/examples/cfd/aero_graph_net/models.py
index 64156249d..01c483ec0 100644
--- a/examples/cfd/aero_graph_net/models.py
+++ b/examples/cfd/aero_graph_net/models.py
@@ -88,6 +88,7 @@ def forward(
         node_features: Tensor,
         edge_features: Tensor,
         graph: Union[DGLGraph, list[DGLGraph], "CuGraphCSC"],
+        **kwargs,
     ) -> Tensor:
         edge_features = self.edge_encoder(edge_features)
         node_features = self.node_encoder(node_features)
diff --git a/modulus/models/meshgraphnet/meshgraphnet.py b/modulus/models/meshgraphnet/meshgraphnet.py
index 9ff65b732..72628efb5 100644
--- a/modulus/models/meshgraphnet/meshgraphnet.py
+++ b/modulus/models/meshgraphnet/meshgraphnet.py
@@ -191,6 +191,7 @@ def forward(
         node_features: Tensor,
         edge_features: Tensor,
         graph: Union[DGLGraph, List[DGLGraph], CuGraphCSC],
+        **kwargs,
     ) -> Tensor:
         edge_features = self.edge_encoder(edge_features)
         node_features = self.node_encoder(node_features)