Fixed edit prop issue