[] fix up tests