corrected a mistake in models