diff --git a/test/tests/test_blocker.lua b/test/tests/test_blocker.lua
index 3191e58..04cb220 100644
--- a/test/tests/test_blocker.lua
+++ b/test/tests/test_blocker.lua
@@ -1,25 +1,26 @@
 return function()
-	local mock_gui = nil
-	local mock_time = nil
-	local mock_input = nil
-	local test_helper = nil
-	local druid_system = nil
-
-	local druid = nil
-
 	describe("Blocker component", function()
+		local mock_gui = nil
+		local mock_time = nil
+		local mock_input = nil
+		local test_helper = nil
+		local druid_system = nil
+
+		local druid = nil ---@type druid.instance
+
 		before(function()
 			mock_gui = require("deftest.mock.gui")
+			mock_gui.mock()
+
 			mock_time = require("deftest.mock.time")
 			mock_input = require("test.helper.mock_input")
 			test_helper = require("test.helper.test_helper")
 			druid_system = require("druid.druid")
 
-			mock_gui.mock()
 			mock_time.mock()
 			mock_time.set(60)
 
-			druid = druid_system.new({})
+			druid = druid_system.new(vmath.vector3())
 		end)
 
 		after(function()
diff --git a/test/tests/test_button.lua b/test/tests/test_button.lua
index 057135b..131522a 100644
--- a/test/tests/test_button.lua
+++ b/test/tests/test_button.lua
@@ -1,22 +1,23 @@
 return function()
-	local mock_gui = nil
-	local mock_time = nil
-	local mock_input = nil
-	local test_helper = nil
-	local druid_system = nil
-
-	local druid = nil
-	local context = nil
-
 	describe("Button Component", function()
+		local mock_gui = nil
+		local mock_time = nil
+		local mock_input = nil
+		local test_helper = nil
+		local druid_system = nil
+
+		local druid = nil
+		local context = nil
+
 		before(function()
 			mock_gui = require("deftest.mock.gui")
+			mock_gui.mock()
+
 			mock_time = require("deftest.mock.time")
 			mock_input = require("test.helper.mock_input")
 			test_helper = require("test.helper.test_helper")
 			druid_system = require("druid.druid")
 
-			mock_gui.mock()
 			mock_time.mock()
 			mock_time.set(60)