Corrected mistake in the code