[commands] Proper cleanup of cogs.

Rapptz · Jan 10, 2016 · efcc55b · efcc55b
1 parent 38bac7f
commit efcc55b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/discord/ext/commands/bot.py b/discord/ext/commands/bot.py
@@ -377,6 +377,8 @@ def remove_cog(self, name):
             if name.startswith('on_'):
                 self.remove_listener(member)
 
+        del cog
+
     # extensions
 
     def load_extension(self, name):
@@ -400,7 +402,7 @@ def unload_extension(self, name):
 
         # remove the cogs registered from the module
         for cogname, cog in self.cogs.copy().items():
-            if cog.__module__ is lib:
+            if inspect.getmodule(cog) is lib:
                 self.remove_cog(cogname)
 
         # first remove all the commands from the module