minor fix in unit tests