diff --git a/control.c b/control.c
index a02b119..acca924 100644
--- a/control.c
+++ b/control.c
@@ -62,7 +62,7 @@
 int control_modified(struct mbox_context *context)
 {
 	/* Flash has been modified - can no longer trust our erased bytemap */
-	flash_set_bytemap(context, 0, context->backend.flash_size,
+	backend_set_bytemap(&context->backend, 0, context->backend.flash_size,
 			    FLASH_DIRTY);
 
 	/* Force daemon to reload all windows -> Set BMC event to notify host */