minor fix for test unit