Small fixes on script.