fixed a small bug

a small fix