corrected mistake in code