Fixed edit in models