diff --git a/tools/markov_import.py b/tools/markov_import.py
index 5270f00..490f47a 100644
--- a/tools/markov_import.py
+++ b/tools/markov_import.py
@@ -4,6 +4,7 @@ import sys
 import re
 
 from omnibot.config import ServerConfig
+from omnibot.bot import Bot
 from plugins.markov import Markov
 
 
@@ -24,10 +25,8 @@ async def main():
     server_config = ServerConfig()
     server_config.load("config.toml")
     # This only works on one plugin per config
-    plugin_config = [
-        plugin for plugin in server_config.plugins if "markov" in plugin["module"]
-    ][0]
-    plugin = Markov(server_config, plugin_config)
+    bot = Bot(server_config)
+    plugin = [plugin for plugin in bot.plugins if isinstance(plugin, Markov)][0]
 
     LINE_RE = re.compile(r"^\[[^\]]+\] <(?P<name>[^>]+)> (?P<message>.+)$")