diff --git a/tests/cd.cpp b/tests/cd.cpp
index 3ba1e2f..29187ff 100644
--- a/tests/cd.cpp
+++ b/tests/cd.cpp
@@ -9,11 +9,11 @@
 #include "trompeloeil_catch.h"
 #include "ast_commands.hpp"
 #include "parser.hpp"
-#include "schema.hpp"
+#include "static_schema.hpp"
 
 TEST_CASE("cd")
 {
-    auto schema = std::make_shared<Schema>();
+    auto schema = std::make_shared<StaticSchema>();
     schema->addContainer("", "a");
     schema->addContainer("", "b");
     schema->addContainer("a", "a2");
diff --git a/tests/leaf_editing.cpp b/tests/leaf_editing.cpp
index 9ae2d97..b8c5591 100644
--- a/tests/leaf_editing.cpp
+++ b/tests/leaf_editing.cpp
@@ -9,11 +9,11 @@
 #include "trompeloeil_catch.h"
 #include "ast_commands.hpp"
 #include "parser.hpp"
-#include "schema.hpp"
+#include "static_schema.hpp"
 
 TEST_CASE("leaf editing")
 {
-    auto schema = std::make_shared<Schema>();
+    auto schema = std::make_shared<StaticSchema>();
     schema->addContainer("", "contA");
     schema->addLeaf("", "leafString", yang::LeafDataTypes::String);
     schema->addLeaf("", "leafDecimal", yang::LeafDataTypes::Decimal);
diff --git a/tests/presence_containers.cpp b/tests/presence_containers.cpp
index 0a678fb..87957cb 100644
--- a/tests/presence_containers.cpp
+++ b/tests/presence_containers.cpp
@@ -10,11 +10,11 @@
 #include "trompeloeil_catch.h"
 #include "ast_commands.hpp"
 #include "parser.hpp"
-#include "schema.hpp"
+#include "static_schema.hpp"
 
 TEST_CASE("presence containers")
 {
-    auto schema = std::make_shared<Schema>();
+    auto schema = std::make_shared<StaticSchema>();
     schema->addContainer("", "a", yang::ContainerTraits::Presence);
     schema->addContainer("", "b");
     schema->addContainer("a", "a2");