Fixed mistake in the code.