Fixing a small bug.