1 files changed, 2 insertions, 2 deletions
diff --git a/tests/helpers/c/main.c b/tests/helpers/c/main.c
index b1dcfcf70..2de29cbdb 100644
--- a/tests/helpers/c/main.c
+++ b/tests/helpers/c/main.c
@@ -24,7 +24,7 @@
 #include "vty.h"
 #include "command.h"
 #include "memory.h"
-#include "memory_vty.h"
+#include "lib_vty.h"
 
 extern void test_init(void);
 
@@ -154,7 +154,7 @@ int main(int argc, char **argv)
 	/* Library inits. */
 	cmd_init(1);
 	vty_init(master, false);
-	memory_init();
+	lib_cmd_init();
 	yang_init();
 	nb_init(master, NULL, 0);