4 files changed, 13 insertions, 5 deletions
diff --git a/lib/io/test/dir_iterator.c b/lib/io/test/dir_iterator.c
index 013e41e..56610b6 100644
--- a/lib/io/test/dir_iterator.c
+++ b/lib/io/test/dir_iterator.c
@@ -40,7 +40,8 @@ int main(int argc, char **argv)
 	(void)argc; (void)argv;
 
 	/* scan the top level hierarchy */
-	dir = dir_iterator_create(TEST_PATH);
+	ret = sqfs_dir_iterator_create_native(&dir, TEST_PATH, 0);
+	TEST_EQUAL_I(ret, 0);
 	TEST_NOT_NULL(dir);
 
 	ret = dir->next(dir, &ent[0]);
@@ -86,7 +87,8 @@ int main(int argc, char **argv)
 		free(ent[i]);
 
 	/* scan first sub hierarchy */
-	dir = dir_iterator_create(TEST_PATH "/dira");
+	ret = sqfs_dir_iterator_create_native(&dir, TEST_PATH "/dira", 0);
+	TEST_EQUAL_I(ret, 0);
 	TEST_NOT_NULL(dir);
 
 	ret = dir->next(dir, &ent[0]);
@@ -132,7 +134,8 @@ int main(int argc, char **argv)
 		free(ent[i]);
 
 	/* scan second sub hierarchy */
-	dir = dir_iterator_create(TEST_PATH "/dirb");
+	ret = sqfs_dir_iterator_create_native(&dir, TEST_PATH "/dirb", 0);
+	TEST_EQUAL_I(ret, 0);
 	TEST_NOT_NULL(dir);
 
 	ret = dir->next(dir, &ent[0]);
@@ -184,7 +187,8 @@ int main(int argc, char **argv)
 		free(ent[i]);
 
 	/* scan first sub hierarchy */
-	dir = dir_iterator_create(TEST_PATH "/dirc");
+	ret = sqfs_dir_iterator_create_native(&dir, TEST_PATH "/dirc", 0);
+	TEST_EQUAL_I(ret, 0);
 	TEST_NOT_NULL(dir);
 
 	ret = dir->next(dir, &ent[0]);
@@ -234,7 +238,8 @@ int main(int argc, char **argv)
 	subb = NULL;
 	subc = NULL;
 
-	dir = dir_iterator_create(TEST_PATH);
+	ret = sqfs_dir_iterator_create_native(&dir, TEST_PATH, 0);
+	TEST_EQUAL_I(ret, 0);
 	TEST_NOT_NULL(dir);
 
 	for (i = 0; i < 5; ++i) {
diff --git a/lib/io/test/dir_tree_iterator.c b/lib/io/test/dir_tree_iterator.c
index 8e11e2a..a6e7fcb 100644
--- a/lib/io/test/dir_tree_iterator.c
+++ b/lib/io/test/dir_tree_iterator.c
@@ -9,6 +9,7 @@
 #include "io/dir_iterator.h"
 #include "sqfs/error.h"
 #include "util/test.h"
+#include "sqfs/io.h"
 #include "compat.h"
 
 static int compare_entries(const void *a, const void *b)
diff --git a/lib/io/test/dir_tree_iterator2.c b/lib/io/test/dir_tree_iterator2.c
index 2833dc5..0ffb86e 100644
--- a/lib/io/test/dir_tree_iterator2.c
+++ b/lib/io/test/dir_tree_iterator2.c
@@ -9,6 +9,7 @@
 #include "io/dir_iterator.h"
 #include "sqfs/error.h"
 #include "util/test.h"
+#include "sqfs/io.h"
 #include "compat.h"
 
 static int compare_entries(const void *a, const void *b)
diff --git a/lib/io/test/dir_tree_iterator3.c b/lib/io/test/dir_tree_iterator3.c
index b82934f..4ac947c 100644
--- a/lib/io/test/dir_tree_iterator3.c
+++ b/lib/io/test/dir_tree_iterator3.c
@@ -9,6 +9,7 @@
 #include "io/dir_iterator.h"
 #include "sqfs/error.h"
 #include "util/test.h"
+#include "sqfs/io.h"
 #include "compat.h"
 
 static int compare_entries(const void *a, const void *b)