diff --git a/docs/kittens/custom.rst b/docs/kittens/custom.rst
index 65f337dab..fc9b575f4 100644
--- a/docs/kittens/custom.rst
+++ b/docs/kittens/custom.rst
@@ -21,6 +21,7 @@ your machine).
 .. code-block:: python
 
     from typing import List
+    from kitty.boss import Boss
 
     def main(args: List[str]) -> str:
         # this is the main entry point of the kitten, it will be executed in
@@ -30,7 +31,7 @@ your machine).
         # handle_result() function
         return answer
 
-    def handle_result(args: List[str], answer: str, target_window_id: int, boss: kitty.boss.Boss) -> None:
+    def handle_result(args: List[str], answer: str, target_window_id: int, boss: Boss) -> None:
         # get the kitty window into which to paste answer
         w = boss.window_id_map.get(target_window_id)
         if w is not None: