fix: поправил модели