diff --git a/src/workbench.lua b/src/workbench.lua
index 9ad5053..b8c3b33 100644
--- a/src/workbench.lua
+++ b/src/workbench.lua
@@ -1,4 +1,5 @@
 local workbench = {}
+WB = {}
 screwdriver = screwdriver or {}
 local min, ceil = math.min, math.ceil
 
@@ -28,7 +29,7 @@ for node, def in pairs(minetest.registered_nodes) do
 end
 
 -- Optionally, you can register custom cuttable nodes in the workbench.
-workbench.custom_nodes_register = {
+WB.custom_nodes_register = {
 	-- "default:leaves",
 }
 
@@ -41,7 +42,7 @@ setmetatable(nodes, {
 	end
 })
 
-nodes = nodes..workbench.custom_nodes_register
+nodes = nodes..WB.custom_nodes_register
 
 -- Nodeboxes definitions.
 workbench.defs = {