diff --git a/tests/test_sorcery_astdb.c b/tests/test_sorcery_astdb.c
index 2371397a4bca011e0084d5a71e47a31e759d9e6c..c45aabe6ddb8b572b616877e079100b9d1d39be3 100644
--- a/tests/test_sorcery_astdb.c
+++ b/tests/test_sorcery_astdb.c
@@ -49,7 +49,7 @@ struct test_sorcery_object {
 /*! \brief Internal function to allocate a test object */
 static void *test_sorcery_object_alloc(const char *id)
 {
-	return ao2_alloc(sizeof(struct test_sorcery_object), NULL);
+	return ast_sorcery_generic_alloc(sizeof(struct test_sorcery_object), NULL);
 }
 
 static struct ast_sorcery *alloc_and_initialize_sorcery(void)
diff --git a/tests/test_sorcery_realtime.c b/tests/test_sorcery_realtime.c
index 6f4196841b4b3f6491465f5a57fb15276aa7d820..85d104988ae7c434be375f8e4f8eb5930b21ad85 100644
--- a/tests/test_sorcery_realtime.c
+++ b/tests/test_sorcery_realtime.c
@@ -201,7 +201,7 @@ struct test_sorcery_object {
 /*! \brief Internal function to allocate a test object */
 static void *test_sorcery_object_alloc(const char *id)
 {
-	return ao2_alloc(sizeof(struct test_sorcery_object), NULL);
+	return ast_sorcery_generic_alloc(sizeof(struct test_sorcery_object), NULL);
 }
 
 static struct ast_sorcery *alloc_and_initialize_sorcery(void)