-Fixing the small bug