test: fix error in unit test