Fix error in script