small bug fix in script