Fixed compile issue (hopefully)