Use new class name in spec