Fix flash class