Fixed compile error (hopefully)