chore(cleanup): fix broken tests