Oops! a bug occured when rewriting code in PEP8, FIXED