Small Fix to script