fixed a small bug 4