Fixed shameful mistake in unit test