update for compile issue @Author : Alex