Fixing a small bug in the logic