fix test by calling proper class