diff --git a/src/tree_data_helpers.c b/src/tree_data_helpers.c
index 41cfcb0..4632d50 100644
--- a/src/tree_data_helpers.c
+++ b/src/tree_data_helpers.c
@@ -584,9 +584,9 @@
                     size_t len = stop - start;
 
                     /* do we already have the prefix? */
-                    mod = ly_type_store_resolve_prefix(ctx, start, len, *format_p, *prefix_data_p);
+                    mod = ly_resolve_prefix(ctx, start, len, *format_p, *prefix_data_p);
                     if (!mod) {
-                        mod = ly_type_store_resolve_prefix(ctx, start, len, format, prefix_data);
+                        mod = ly_resolve_prefix(ctx, start, len, format, prefix_data);
                         if (mod) {
                             assert(*format_p == LY_PREF_SCHEMA_RESOLVED);
                             /* store a new prefix - module pair */