diff --git a/data/prodinf01n04.main.unkin.net.yaml b/data/prodinf01n04.main.unkin.net.yaml
index 4047e86..59ce265 100644
--- a/data/prodinf01n04.main.unkin.net.yaml
+++ b/data/prodinf01n04.main.unkin.net.yaml
@@ -1,4 +1,4 @@
 ---
 classes:
-  - roles::puppet::puppetdb
+  - roles::puppet::puppetdb_api
 environment: testing
diff --git a/data/prodinf01n05.main.unkin.net.yaml b/data/prodinf01n05.main.unkin.net.yaml
index 5cb2820..cf3988b 100644
--- a/data/prodinf01n05.main.unkin.net.yaml
+++ b/data/prodinf01n05.main.unkin.net.yaml
@@ -1,4 +1,4 @@
 ---
 classes:
-  - roles::base
+  - roles::puppet::puppetdb_sql
 environment: testing