diff --git a/models/dataset.go b/models/dataset.go
index af47c53fe5..c0d82d2507 100755
--- a/models/dataset.go
+++ b/models/dataset.go
@@ -348,7 +348,7 @@ func GetDatasetByRepo(repo *Repository) (*Dataset, error) {
 	if has {
 		return dataset, nil
 	} else {
-		return nil, errors.New("Not Found")
+		return nil, ErrNotExist{repo.ID}
 	}
 }
 
diff --git a/models/repo.go b/models/repo.go
index b5d4921e4b..25bfb4a74e 100755
--- a/models/repo.go
+++ b/models/repo.go
@@ -1607,14 +1607,16 @@ func updateRepository(e Engine, repo *Repository, visibilityChanged bool) (err e
 			}
 
 			dataset, err := GetDatasetByRepo(repo)
-			if err != nil {
+			if err != nil && !IsErrNotExist(err) {
 				return err
 			}
-			_, err = e.Where("dataset_id = ?", dataset.ID).Cols("is_private").Update(&Attachment{
-				IsPrivate: true,
-			})
-			if err != nil {
-				return err
+			if dataset != nil {
+				_, err = e.Where("dataset_id = ?", dataset.ID).Cols("is_private").Update(&Attachment{
+					IsPrivate: true,
+				})
+				if err != nil {
+					return err
+				}
 			}
 
 		} else {