diff --git a/aurora/configuration/commands.py b/aurora/configuration/commands.py index 0d1e5ef..a563b6c 100644 --- a/aurora/configuration/commands.py +++ b/aurora/configuration/commands.py @@ -1,6 +1,8 @@ from redbot.core import commands from redbot.core.utils.chat_formatting import error, warning +from aurora.configuration.menus.addrole import Addrole +from aurora.configuration.menus.immune import Immune from aurora.configuration.menus.overrides import Overrides from aurora.configuration.embed import addrole, overrides, immune, guild from aurora.abc import Mixin @@ -38,14 +40,14 @@ class Configuration(Mixin): """Manage the addrole whitelist. Roles added to this list are also applied to `/removerole`.""" - await ctx.send(embed=await addrole(ctx), view=await addrole(ctx)) + await ctx.send(embed=await addrole(ctx), view=Addrole(ctx)) @aurora_settings.command(name="immunity") @commands.admin_or_permissions(manage_guild=True) @commands.guild_only() async def aurora_settings_immunity(self, ctx: commands.Context): """Manage the immunity whitelist.""" - await ctx.send(embed=await immune(ctx), view=await immune(ctx)) + await ctx.send(embed=await immune(ctx), view=Immune(ctx)) @aurora.group(autohelp=True, name="import") @commands.admin()