fixup! fix failing unit-test