Fix dumb mistake in test case