Fix the flash problem