Fixed mistake in model