Fixed small compilation error