From 8c3f87b02984ad14545fc2361aa6fa856ad4debb Mon Sep 17 00:00:00 2001
From: jesopo <github@lolnerd.net>
Date: Sun, 9 Dec 2018 11:22:30 +0000
Subject: [PATCH] change `self.modules` to be a dict of str:LoadedModule, not
 str:BaseModule

---
 src/ModuleManager.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ModuleManager.py b/src/ModuleManager.py
index 2b1cea58..0d4cf887 100644
--- a/src/ModuleManager.py
+++ b/src/ModuleManager.py
@@ -61,7 +61,7 @@ class ModuleManager(object):
         self.log = log
         self.directory = directory
 
-        self.modules = {} # type: typing.Dict[str, BaseModule]
+        self.modules = {} # type: typing.Dict[str, LoadedModule]
         self.waiting_requirement = {} # type: typing.Dict[str, typing.Set[str]]
 
     def list_modules(self) -> typing.List[str]:
@@ -146,7 +146,7 @@ class ModuleManager(object):
                 [name, str(e)])
             raise
 
-        self.modules[loaded_module.name] = loaded_module.module
+        self.modules[loaded_module.name] = loaded_module
         if loaded_module.name in self.waiting_requirement:
             for requirement_name in self.waiting_requirement[
                     loaded_module.name]: