diff --git a/test/in_memory_catalog_test.cc b/test/in_memory_catalog_test.cc
index 1fe598a2c..77b0823e1 100644
--- a/test/in_memory_catalog_test.cc
+++ b/test/in_memory_catalog_test.cc
@@ -25,7 +25,7 @@
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
-#include "iceberg/arrow/arrow_fs_file_io.h"
+#include "iceberg/arrow/arrow_fs_file_io_internal.h"
 #include "iceberg/table.h"
 #include "iceberg/table_metadata.h"
 #include "matchers.h"
diff --git a/test/manifest_reader_test.cc b/test/manifest_reader_test.cc
index 6821ae79e..2fb20b73b 100644
--- a/test/manifest_reader_test.cc
+++ b/test/manifest_reader_test.cc
@@ -22,7 +22,7 @@
 #include <arrow/filesystem/localfs.h>
 #include <gtest/gtest.h>
 
-#include "iceberg/arrow/arrow_fs_file_io.h"
+#include "iceberg/arrow/arrow_fs_file_io_internal.h"
 #include "iceberg/avro/avro_reader.h"
 #include "iceberg/avro/avro_register.h"
 #include "iceberg/avro/avro_schema_util_internal.h"