Minor changes

Revise a small bug