Small error in main script.