fixing another small bug