diff --git a/src/bonsai/bonsai/bim/schema.py b/src/bonsai/bonsai/bim/schema.py
index 63c03e4a4a..7890f90006 100644
--- a/src/bonsai/bonsai/bim/schema.py
+++ b/src/bonsai/bonsai/bim/schema.py
@@ -20,8 +20,9 @@ import ifcopenshell
 import ifcopenshell.util.pset
 import bonsai.tool as tool
 import bpy
+from typing import TYPE_CHECKING
 
-if bpy.app.version >= (5, 0, 0):
+if TYPE_CHECKING or bpy.app.version >= (5, 0, 0):
     import _bpy_restrict_state as bpy_restrict_state
 else:
     import bpy_restrict_state