diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
index 4902756..2963eba 100644
--- a/meshgpt_pytorch/meshgpt_pytorch.py
+++ b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -1603,7 +1603,7 @@ def forward_on_codes(
             )
 
             if exists(self.coarse_post_gateloop_block):
-                face_codes, coarse_post_gateloop_cache = self.coarse_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)
+                face_codes, coarse_post_gateloop_cache = self.coarse_post_gateloop_block(face_codes, cache = coarse_post_gateloop_cache)
 
         else:
             attended_face_codes = None
diff --git a/meshgpt_pytorch/version.py b/meshgpt_pytorch/version.py
index 77f1c8e..51ed7c4 100644
--- a/meshgpt_pytorch/version.py
+++ b/meshgpt_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.5.0'
+__version__ = '1.5.1'