diff --git a/models.py b/models.py
index a2fb3f8..05d6a2c 100644
--- a/models.py
+++ b/models.py
@@ -596,7 +596,7 @@ class FouriER(torch.nn.Module):
             graph = degree @ graph @ degree
 
         for idx, block in enumerate(self.network):
-            x = block(x)
+            x = block(x, graph)
         # output only the features of last layer for image classification
         return x