diff --git a/examples/research_projects/lxmert/extracting_data.py b/examples/research_projects/lxmert/extracting_data.py
index 9790e20ad86b..38d7a43ae72d 100644
--- a/examples/research_projects/lxmert/extracting_data.py
+++ b/examples/research_projects/lxmert/extracting_data.py
@@ -98,7 +98,7 @@ def file_generator(self):
     def __call__(self):
         # make writer
         if not TEST:
-            writer = datasets.ArrowWriter(features=self.schema, path=self.outputfile)
+            writer = datasets.arrow_writer.ArrowWriter(features=self.schema, path=self.outputfile)
         # do file generator
         for i, (img_ids, filepaths) in enumerate(self.file_generator):
             images, sizes, scales_yx = self.preprocess(filepaths)
diff --git a/examples/research_projects/visual_bert/extracting_data.py b/examples/research_projects/visual_bert/extracting_data.py
index 9790e20ad86b..38d7a43ae72d 100644
--- a/examples/research_projects/visual_bert/extracting_data.py
+++ b/examples/research_projects/visual_bert/extracting_data.py
@@ -98,7 +98,7 @@ def file_generator(self):
     def __call__(self):
         # make writer
         if not TEST:
-            writer = datasets.ArrowWriter(features=self.schema, path=self.outputfile)
+            writer = datasets.arrow_writer.ArrowWriter(features=self.schema, path=self.outputfile)
         # do file generator
         for i, (img_ids, filepaths) in enumerate(self.file_generator):
             images, sizes, scales_yx = self.preprocess(filepaths)