diff --git a/global/overlay/etc/puppet/setup_cosmos_modules b/global/overlay/etc/puppet/setup_cosmos_modules
index 3029e10..acc8f2a 100755
--- a/global/overlay/etc/puppet/setup_cosmos_modules
+++ b/global/overlay/etc/puppet/setup_cosmos_modules
@@ -85,6 +85,11 @@ def main():
             "upgrade": "yes",
             "tag": "stable-2023v1-2*",
         },
+        "augeas": {
+            "repo": "https://github.com/SUNET/puppet-augeas.git",
+            "upgrade": "yes",
+            "tag": "sunet-2*",
+        },
         "concat": {
             "repo": "https://github.com/SUNET/puppetlabs-concat.git",
             "upgrade": "yes",
@@ -115,17 +120,18 @@ def main():
             OS_INFO["ID"] == "ubuntu"
             and debian_support.version_compare(OS_INFO["VERSION_ID"], "24.04") >= 0
         ):
+            del modules["augeas"]
             del modules["apt"]
             del modules["concat"]
             del modules["stdlib"]
 
-        # Make test machines use special test branch of puppet-sunet:
-        if OS_INFO["ID"] == "ubuntu":
-            nodename = platform.node()
-            name_parts = nodename.split("-")
-            if len(name_parts) > 1:
-                if name_parts[1] == "test":
-                    modules["sunet"]["tag"] = "testing-2*"
+#        # Make test machines use special test branch of puppet-sunet:
+#        if OS_INFO["ID"] == "ubuntu":
+#            nodename = platform.node()
+#            name_parts = nodename.split("-")
+#            if len(name_parts) > 1:
+#                if name_parts[1] == "test":
+#                    modules["sunet"]["tag"] = "testing-2*"
 
     # Build list of expected file content
     file_lines = create_file_content(modules)