fixing a simple mistake in the code