Fixed test by correcting pagination class