Corrected a mistake in code.