Fixed a mistake in the code.