fix error in unit test