Fixed another small bug