minor fix to unit test.'