A wheel assembly for a shaft of a screening machine includes a disc and a first and a second weight unit. The first and the second weight units are arranged such that the disc is in equilibrium along a longitudinal axis passing through the center of the disc and the longitudinal axis is perpendicular to a diametrical extension of the disc. A screening machine having the wheel assembly is also provided.