Minnor ammend in main script.