fixed the flash_class issue