Скорректировано поведение, удален трейт и поправлена работа со связями.