ok... _actually_ fix up tests