Fixed small issue with compiling