fixing compile bug