fix a small bug

.