small mistake in my head. fix test case