FIxing a small bug