just fixed a small bug