Fixed compile bug