minor fix to unittest