Fix small compilation errors