Fix small mistake in test generation