Fixing one small bug.