fix busted tests