Fixed small compilation issue