Fix flash issue