fix class names in unittests