Fixed small compilation errors