Hopefuly fix compile bug