fixed a small bug in script