fix loading class bug