1 files changed, 8 insertions, 2 deletions
diff --git a/BuildTools/FixIncludes.py b/BuildTools/FixIncludes.py
index 0780a5b..d409a2c 100755
--- a/BuildTools/FixIncludes.py
+++ b/BuildTools/FixIncludes.py
@@ -19,7 +19,7 @@ c_stdlib_headers = Set(["assert.h",  "limits.h",  "signal.h",  "stdlib.h", "ctyp
 cpp_stdlib_headers = Set(["algorithm", "fstream", "list", "regex", "typeindex", "array", "functional", "locale", "set", "typeinfo", "atomic", "future", "map", "sstream", "type_traits", "bitset", "initializer_list", "memory", "stack", "unordered_map", "chrono", "iomanip", "mutex", "stdexcept", "unordered_set", "codecvt", "ios", "new", "streambuf", "utility", "complex", "iosfwd", "numeric", "string", "valarray", "condition_variable", "iostream", "ostream", "strstream", "vector", "deque", "istream", "queue", "system_error", "exception", "iterator", "random", "thread", "forward_list", "limits", "ratio", "tuple", "cassert", "ciso646", "csetjmp", "cstdio", "ctime", "cctype", "climits", "csignal", "cstdlib", "cwchar", "cerrno", "clocale", "cstdarg", "cstring", "cwctype", "cfloat", "cmath", "cstddef"])
 
 class HeaderType:
-		PRAGMA_ONCE, CORRESPONDING_HEADER, C_STDLIB, CPP_STDLIB, BOOST, QT, OTHER, SWIFTEN, SWIFT_CONTROLLERS, SLUIFT, SWIFTOOLS, SWIFT = range(12)
+		PRAGMA_ONCE, CORRESPONDING_HEADER, C_STDLIB, CPP_STDLIB, BOOST, QT, OTHER, SWIFTEN, LIMBER, SLIMBER, SWIFT_CONTROLLERS, SLUIFT, SWIFTOOLS, SWIFT = range(14)
 
 def findHeaderBlock(lines):
 	start = False
@@ -46,7 +46,7 @@ def lineToFileName(line):
 	return False
 
 def fileNameToHeaderType(name):
-	if name.endswith(filename_name + ".h"):
+	if name.endswith("/" + filename_name + ".h"):
 		return HeaderType.CORRESPONDING_HEADER
 
 	if name in c_stdlib_headers:
@@ -64,6 +64,12 @@ def fileNameToHeaderType(name):
 	if name.startswith("Swiften"):
 		return HeaderType.SWIFTEN
 
+	if name.startswith("Limber"):
+		return HeaderType.LIMBER
+
+	if name.startswith("Slimber"):
+		return HeaderType.SLIMBER
+
 	if name.startswith("Swift/Controllers"):
 		return HeaderType.SWIFT_CONTROLLERS