fix class name for tests