diff --git a/tests/utests/basic/test_xpath.c b/tests/utests/basic/test_xpath.c
index 0da6f7b..8be1fbf 100644
--- a/tests/utests/basic/test_xpath.c
+++ b/tests/utests/basic/test_xpath.c
@@ -299,20 +299,19 @@
     assert_non_null(mod);
 
     /* some random paths just making sure the API function works */
-    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(NULL, mod->compiled->data, "/a:*", 0, &set));
+    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(UTEST_LYCTX, NULL, "/a:*", 0, &set));
     assert_int_equal(4, set->count);
 
     ly_set_free(set, NULL);
 
     /* all nodes from all modules (including internal, which can change easily, so check just the test modules) */
-    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(NULL, mod->compiled->data, "//.", 0, &set));
+    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(UTEST_LYCTX, NULL, "//.", 0, &set));
     assert_in_range(set->count, 16, UINT32_MAX);
 
     ly_set_free(set, NULL);
 
-    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(NULL, mod->compiled->data->next->next, "/a:c/ll[a='val1']/ll[a='val2']/b",
-            0, &set));
-    assert_int_equal(7, set->count);
+    assert_int_equal(LY_SUCCESS, lys_find_xpath_atoms(UTEST_LYCTX, NULL, "/a:c/ll[a='val1']/ll[a='val2']/b", 0, &set));
+    assert_int_equal(6, set->count);
 
     ly_set_free(set, NULL);
 }