corrected an error in __main__