The wheel assembly has three bearings, one on each side of he wheel itself and one in the cush drive assembly. It's the bearing in the cush drive assembly that would give the side to side movement.
However, if it were me and the cush drive bearing was ready for replacement, I replace all three bearings.