Fixing another small bug.