diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py
index 87bb9e07dc..ad00147ab3 100644
--- a/src/diffusers/models/attention_processor.py
+++ b/src/diffusers/models/attention_processor.py
@@ -1182,7 +1182,6 @@ class AuraFlowAttnProcessor2_0:
         attn: Attention,
         hidden_states: torch.FloatTensor,
         encoder_hidden_states: torch.FloatTensor = None,
-        i=0,
         *args,
         **kwargs,
     ) -> torch.FloatTensor:
diff --git a/tests/pipelines/aura_flow/test_pipeline_aura_dlow.py b/tests/pipelines/aura_flow/test_pipeline_aura_flow.py
similarity index 100%
rename from tests/pipelines/aura_flow/test_pipeline_aura_dlow.py
rename to tests/pipelines/aura_flow/test_pipeline_aura_flow.py