diff --git a/tests/pglabtests/tst_CodeBuilder.cpp b/tests/pglabtests/tst_CodeBuilder.cpp
index cf7d349..5f9fcb2 100644
--- a/tests/pglabtests/tst_CodeBuilder.cpp
+++ b/tests/pglabtests/tst_CodeBuilder.cpp
@@ -2,8 +2,8 @@
 #include <gmock/gmock-matchers.h>
 #include "PrintTo_Qt.h"
 #include <QRegularExpression>
-#include "PgType.h"
-#include "PgTypeContainer.h"
+#include "catalog/PgType.h"
+#include "catalog/PgTypeContainer.h"
 #include "codebuilder/IndentationConfig.h"
 #include "codebuilder/TypeMappings.h"
 #include "Pgsql_oids.h"
diff --git a/tests/pglabtests/tst_TypeMappings.cpp b/tests/pglabtests/tst_TypeMappings.cpp
index 2ff7102..9735936 100644
--- a/tests/pglabtests/tst_TypeMappings.cpp
+++ b/tests/pglabtests/tst_TypeMappings.cpp
@@ -1,8 +1,8 @@
 #include <gtest/gtest.h>
 #include <gmock/gmock-matchers.h>
 //#include <QRegularExpression>
-#include "PgType.h"
-#include "PgTypeContainer.h"
+#include "catalog/PgType.h"
+#include "catalog/PgTypeContainer.h"
 //#include "PrintTo_Qt.h"
 //#include "codebuilder/IndentationConfig.h"
 #include "codebuilder/TypeMappings.h"