Slight change, so the script is more robust.