update test to matchup with class