Corrected mistake in models

Corrected mistake in models