Fix __main__.