Fixing one small bug