Improve typing for Plugin init

1 files changed, 2 insertions, 1 deletions

diff --git a/asciidoc/plugin.py b/asciidoc/plugin.py
index 87e50f9..765019b 100644
--- a/asciidoc/plugin.py
+++ b/asciidoc/plugin.py
@@ -5,6 +5,7 @@ import sys
 from typing import NoReturn
 import zipfile
 
+from .message import Message
 from .utils import userdir
 
 
@@ -14,7 +15,7 @@ class Plugin:
     """
     CMDS = ('install', 'remove', 'list', 'build')
 
-    def __init__(self, type: str, message, config):
+    def __init__(self, type: str, message: Message, config):
         self.type = type
         self.message = message
         self.config = config