Fixed small mistake in unit test