diff --git a/test/erase_flash.c b/test/erase_flash.c
index 1a707a2..f4f4821 100644
--- a/test/erase_flash.c
+++ b/test/erase_flash.c
@@ -109,7 +109,7 @@
 	n_ioctls = 0;
 	recorded = NULL;
 
-	init_flash_dev(&context);
+	flash_dev_init(&context);
 
 	/* Erase from an unknown state */
 	rc = erase_flash(&context, 0, sizeof(data));
diff --git a/test/mbox.c b/test/mbox.c
index 7e03292..b45f42f 100644
--- a/test/mbox.c
+++ b/test/mbox.c
@@ -242,7 +242,7 @@
 	assert(rc == 0);
 	test.context.fds[MBOX_FD].fd = test.mbox.fd;
 
-	rc = init_flash_dev(&test.context);
+	rc = flash_dev_init(&test.context);
 	assert(rc == 0);
 
 	rc = fallocate(test.flash.fd, 0, 0, test.context.mtd_info.size);
diff --git a/test/write_flash.c b/test/write_flash.c
index b09e7de..4f37bef 100644
--- a/test/write_flash.c
+++ b/test/write_flash.c
@@ -63,7 +63,7 @@
 
 	mbox_vlog = &mbox_log_console;
 
-	rc = init_flash_dev(context);
+	rc = flash_dev_init(context);
 	assert(rc == 0);
 
 	map = mmap(NULL, MEM_SIZE, PROT_READ, MAP_PRIVATE, tmp->fd, 0);