FIxed small bug