Just fixed a small bug