Fix another small bug