Bug fix

jtyliu · Mar 12, 2022 · 5f1a4d4 · 5f1a4d4
1 parent 0cfb944
commit 5f1a4d4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/extensions/handles.py b/extensions/handles.py
@@ -146,9 +146,12 @@ async def link(ctx: lightbulb.Context) -> None:
     username = ctx.options.username
     # Check if user exists
     query = Query()
-    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