diff --git a/src/yacs_stubgen/__init__.py b/src/yacs_stubgen/__init__.py
index f3277b6..ff72c26 100644
--- a/src/yacs_stubgen/__init__.py
+++ b/src/yacs_stubgen/__init__.py
@@ -1,5 +1,4 @@
 from pathlib import Path
-from types import NoneType
 from typing import Any, Dict, Optional, Union
 
 import yaml
@@ -8,7 +7,7 @@
 typo_map = {
     list: "T.Sequence",
     tuple: "T.Sequence",
-    NoneType: "T.Any",
+    type(None): "T.Any",
 }