Fix find class bug