fix bug about class of python