Fix small compilation error