Fixed casting results to their correct models