diff --git a/build/scripts/tools/unittests.lua b/build/scripts/tools/unittests.lua
index 5dcf6e215..6c44540a3 100644
--- a/build/scripts/tools/unittests.lua
+++ b/build/scripts/tools/unittests.lua
@@ -14,7 +14,9 @@ TOOL.Includes = {
 
 TOOL.Files = {
 	"../tests/main.cpp",
+	"../tests/Engine/**.hpp",
 	"../tests/Engine/**.cpp",
+	"../tests/SDK/**.hpp",
 	"../tests/SDK/**.cpp"
 }
 
diff --git a/build/scripts/tools/unittests_server.lua b/build/scripts/tools/unittests_server.lua
index 44efbff84..c94a54d6d 100644
--- a/build/scripts/tools/unittests_server.lua
+++ b/build/scripts/tools/unittests_server.lua
@@ -15,7 +15,9 @@ TOOL.Includes = {
 
 TOOL.Files = {
 	"../tests/main.cpp",
+	"../tests/Engine/**.hpp",
 	"../tests/Engine/**.cpp",
+	"../tests/SDK/**.hpp",
 	"../tests/SDK/**.cpp"
 }