minor update in unit tests