Fixed issue #456 (#457)

cheran-senthil · Jul 18, 2021 · 2c0e2e8 · 2c0e2e8
1 parent 7093a56
commit 2c0e2e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tle/util/db/user_db_conn.py b/tle/util/db/user_db_conn.py
@@ -475,11 +475,11 @@ def add_starboard_message(self, original_msg_id, starboard_msg_id, guild_id):
         self.conn.commit()
 
     def check_exists_starboard_message(self, original_msg_id):
-        query = ('SELECT 1 '
+        query = ('SELECT COUNT(*) AS cnt '
                  'FROM starboard_message '
                  'WHERE original_msg_id = ?')
         res = self.conn.execute(query, (original_msg_id,)).fetchone()
-        return res is not None
+        return res.cnt != 0
 
     def remove_starboard_message(self, *, original_msg_id=None, starboard_msg_id=None):
         assert (original_msg_id is None) ^ (starboard_msg_id is None)