fixup! fix bugs