Script bug fix!