I fixed the selection class updating.