diff --git a/src/tracerestrict.cpp b/src/tracerestrict.cpp
index 975d79ee42..3aa26935bc 100644
--- a/src/tracerestrict.cpp
+++ b/src/tracerestrict.cpp
@@ -21,6 +21,9 @@
 #include "group.h"
 #include "string_func.h"
 #include "pathfinder/yapf/yapf_cache.h"
+
+#include "safeguards.h"
+
 #include <vector>
 #include <algorithm>
 
diff --git a/src/tracerestrict_gui.cpp b/src/tracerestrict_gui.cpp
index f9334b03f7..5bc9b5fe05 100644
--- a/src/tracerestrict_gui.cpp
+++ b/src/tracerestrict_gui.cpp
@@ -44,6 +44,8 @@
 #include "table/sprites.h"
 #include "core/geometry_func.hpp"
 
+#include "safeguards.h"
+
 /** Widget IDs */
 enum TraceRestrictWindowWidgets {
 	TR_WIDGET_CAPTION,