minor fix to unit test