diff --git a/zulip_bots/zulip_bots/zulip_bot_output.py b/zulip_bots/zulip_bots/zulip_bot_output.py
index 7fc6c6b..9d60d87 100644
--- a/zulip_bots/zulip_bots/zulip_bot_output.py
+++ b/zulip_bots/zulip_bots/zulip_bot_output.py
@@ -2,19 +2,16 @@
 from __future__ import print_function
 from __future__ import absolute_import
 
-import argparse
-import sys
 import os
-from types import ModuleType
-from importlib import import_module
-from os.path import basename, splitext
+import sys
+import argparse
+import zulip_bots
 
-import six
 from six.moves import configparser
-import mock
-from mock import MagicMock, patch
 
-from zulip_bots.lib import run_message_handler_for_bot, StateHandler
+from mock import MagicMock, patch
+from zulip_bots.lib import StateHandler
+from zulip_bots.lib import ExternalBotHandler
 from zulip_bots.provision import provision_bot
 from zulip_bots.run import import_module_from_source, name_and_path_match