fix a mistake in the code