diff --git a/pkg/models/favorites.go b/pkg/models/favorites.go
index 7aa093df7..f04a685ab 100644
--- a/pkg/models/favorites.go
+++ b/pkg/models/favorites.go
@@ -66,7 +66,7 @@ func removeFromFavorite(s *xorm.Session, entityID int64, a web.Auth, kind Favori
 	}
 
 	_, err = s.
-		Where("task_id = ? AND user_id = ? AND kind = ?", entityID, u.ID, kind).
+		Where("entity_id = ? AND user_id = ? AND kind = ?", entityID, u.ID, kind).
 		Delete(&Favorite{})
 	return err
 }
@@ -79,7 +79,7 @@ func isFavorite(s *xorm.Session, entityID int64, a web.Auth, kind FavoriteKind)
 	}
 
 	return s.
-		Where("task_id = ? AND user_id = ? AND kind = ?", entityID, u.ID, kind).
+		Where("entity_id = ? AND user_id = ? AND kind = ?", entityID, u.ID, kind).
 		Exist(&Favorite{})
 }