Small correction to script