minor fix for 2 unit test