Fix flash write command tests.