little mistake in code