diff --git a/examples/data2vec/models/data2vec_audio.py b/examples/data2vec/models/data2vec_audio.py
index 5cf5b53..ccaf4a4 100644
--- a/examples/data2vec/models/data2vec_audio.py
+++ b/examples/data2vec/models/data2vec_audio.py
@@ -423,7 +423,7 @@ class Data2VecAudioModel(BaseFairseqModel):
 
         # extracting both the source and the augmented audios from source
         source_audios = source[0]
-        aug_audios = source[1]
+        aug_audios = source[0]
 
         if self.feature_grad_mult > 0:
             features = self.feature_extractor(source_audios) #features
diff --git a/setup.py b/setup.py
index a7ce61a..3dc7b44 100644
--- a/setup.py
+++ b/setup.py
@@ -105,12 +105,12 @@ try:
                     "fairseq/clib/libnat/edit_dist.cpp",
                 ],
             ),
-            cpp_extension.CppExtension(
-                "alignment_train_cpu_binding",
-                sources=[
-                    "examples/operators/alignment_train_cpu.cpp",
-                ],
-            ),
+            # cpp_extension.CppExtension(
+            #     "alignment_train_cpu_binding",
+            #     sources=[
+            #         "examples/operators/alignment_train_cpu.cpp",
+            #     ],
+            # ),
         ]
     )
     if "CUDA_HOME" in os.environ: