Fixed error in models