Fix sys.exit() error