minor fix to issue in UntiTest