diff --git a/src/bonsai/bonsai/bim/schema.py b/src/bonsai/bonsai/bim/schema.py
index 6bc97f6ae7..63c03e4a4a 100644
--- a/src/bonsai/bonsai/bim/schema.py
+++ b/src/bonsai/bonsai/bim/schema.py
@@ -20,7 +20,11 @@ import ifcopenshell
 import ifcopenshell.util.pset
 import bonsai.tool as tool
 import bpy
-import bpy_restrict_state
+
+if bpy.app.version >= (5, 0, 0):
+    import _bpy_restrict_state as bpy_restrict_state
+else:
+    import bpy_restrict_state
 
 
 class IfcSchema: