fix mistake in code