diff --git a/py/nodes/sageAttention.py b/py/nodes/sageAttention.py
index 755e0fd..af1aa46 100644
--- a/py/nodes/sageAttention.py
+++ b/py/nodes/sageAttention.py
@@ -155,7 +155,7 @@ def attention_bleh(  # noqa: PLR0914
         result = result.transpose(1, 2)
     if not skip_output_reshape:
         result = result.reshape(batch, -1, heads * dim_head)
-    return result
+    return result.contiguous()
 
 
 def copy_funattrs(fun, dest=None):