fix mistake in running code