just cleared code by pycharm
small bug fix