Fixed small compile error