diff --git a/tests/torchtune/modules/_export/test_export_position_embeddings.py b/tests/torchtune/modules/_export/test_export_position_embeddings.py
index 6907ca3edd..3beb23e7ef 100644
--- a/tests/torchtune/modules/_export/test_export_position_embeddings.py
+++ b/tests/torchtune/modules/_export/test_export_position_embeddings.py
@@ -161,7 +161,6 @@ def test_tiled_token_positional_embedding_aoti(self):
         with tempfile.TemporaryDirectory() as tmpdir:
             path = torch._inductor.aoti_compile_and_package(
                 tpe_ep,
-                (self.x, self.aspect_ratio),
                 package_path=os.path.join(tmpdir, "tpe.pt2"),
             )
             tpe_aoti = load_package(path)