Corrects little mistake in the code