Bug Fix in handle.py

jtyliu · Mar 12, 2022 · 0cfb944 · 0cfb944
1 parent ffaf5a8
commit 0cfb944
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/extensions/handles.py b/extensions/handles.py
@@ -213,9 +213,12 @@ async def _set(ctx):
     query = Query()
 
     if username != "+remove":
-        user = await query.get_user(username)
+        try:
+            user = await query.get_user(username)
 
-        if user is None:
+            if user is None:
+                raise ObjectNotFound()
+        except ObjectNotFound:
             await ctx.respond(escape_markdown(f"{username} does not exist on dmoj"))
             return