fixup! Fix a bug