diff --git a/test/erase_flash.c b/test/erase_flash.c
index f4f4821..eae2627 100644
--- a/test/erase_flash.c
+++ b/test/erase_flash.c
@@ -204,7 +204,7 @@
 	recorded = NULL;
 	n_ioctls = 0;
 
-	free_flash_dev(&context);
+	flash_dev_free(&context);
 
 	return rc;
 }
diff --git a/test/write_flash.c b/test/write_flash.c
index 4f37bef..f0c107c 100644
--- a/test/write_flash.c
+++ b/test/write_flash.c
@@ -99,7 +99,7 @@
 	rc = memcmp(src, map, sizeof(src));
 	assert(rc == 0);
 
-	free_flash_dev(context);
+	flash_dev_free(context);
 
 	return rc;
 }