Fix error in unit test