[] fix up failing tests