diff --git a/stubs/yaml.pyi b/stubs/yaml.pyi
index 4f231f5..4e03d09 100644
--- a/stubs/yaml.pyi
+++ b/stubs/yaml.pyi
@@ -1,4 +1,4 @@
-from typing import Union, List, Dict, Any, IO
+from typing import Union, List, Dict, Any
 
 
 Basic = Union[List, Dict[str, Any]]
@@ -9,6 +9,7 @@
 class CLoader: ...
 class CDumper: ...
 
-# def load(stream: IO, loader: Any) -> Any: ...
 def load(data: bytes, Loader: Any = None, encoding: str = 'utf8') -> Any: ...
-def dump(data: str, Dumper: Any = None, encoding: str = 'utf8') -> bytes: ...
+def dump(data: Any, Dumper: Any = None, encoding: str = 'utf8') -> bytes: ...
+def safe_load(data: bytes, encoding: str = 'utf8') -> Any: ...
+def safe_dump(data: Any, encoding: str = 'utf8') -> bytes: ...