diff --git a/toml/decoder.pyi b/toml/decoder.pyi
index 967d3dd..cc9e416 100644
--- a/toml/decoder.pyi
+++ b/toml/decoder.pyi
@@ -1,5 +1,5 @@
 from toml.tz import TomlTz as TomlTz
-from typing import Any, Optional
+from typing import Any, IO, MutableMapping, Optional, Type, Union
 
 unicode = str
 basestring = str
diff --git a/toml/encoder.pyi b/toml/encoder.pyi
index 194a358..19600bf 100644
--- a/toml/encoder.pyi
+++ b/toml/encoder.pyi
@@ -1,5 +1,5 @@
 from toml.decoder import InlineTableDict as InlineTableDict
-from typing import Any, Optional
+from typing import Any, IO, Mapping, Optional
 
 unicode = str