fix a typo for the main function