fix up __main__