diff --git a/writeprotect.c b/writeprotect.c
index 302451b..acc8714 100644
--- a/writeprotect.c
+++ b/writeprotect.c
@@ -169,7 +169,7 @@
 
 		value = (value & ~write_masks[reg]) | expected;
 
-		if (spi_write_register(flash, reg, value, WRSR_EITHER))
+		if (spi_write_register(flash, reg, value, default_wrsr_target(flash)))
 			return FLASHPROG_WP_ERR_WRITE_FAILED;
 	}