fixed a small bug 3