fixed a small bug 2