Fixed: class name for tests