fix __main__