fix unit tests, again, missed this one!