update test to not just look for the classname