Создание подкласса и переопределение функции генератора в python

Мне нужно переопределить метод родительского класса, который является генератором, и мне интересно, как это сделать. Что-то не так в следующем или более эффективном способе?

class A:
    def gen(self):
        yield 1
        yield 2

class B(A):
    def gen(self):
        yield 3
        for n in super().gen():
            yield n