diff --git a/extensions/handles.py b/extensions/handles.py index fb2d4fd..72d50e8 100644 --- 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