Minor script correction