Update class unit test to remove stale parameters