Avoid class name mismatch between tests