diff --git a/fedora-atomic.conf b/fedora-atomic.conf
index eb1f914..fd02d6d 100644
--- a/fedora-atomic.conf
+++ b/fedora-atomic.conf
@@ -262,20 +262,14 @@ ostree_installer = [
                                   "https://kojipkgs.fedoraproject.org/pub/fedora/linux/releases/26/Everything/$arch/os/"],
             "release": None,
             "installpkgs": ["fedora-productimg-atomic"],
-            "add_template": ["atomic-installer/lorax-configure-repo.tmpl"],
+            "rootfs_size": "3",
+            "add_template": ["atomic-installer/lorax-configure-repo.tmpl",
+                             "atomic-installer/lorax-embed-repo.tmpl"],
             "add_template_var": [
                 "ostree_repo=https://kojipkgs.fedoraproject.org/atomic/26/",
                 "ostree_osname=fedora-atomic",
-                "ostree_install_ref=fedora/26/x86_64/updates/atomic-host",
-                "ostree_update_ref=fedora/26/x86_64/atomic-host",
+                "ostree_ref=fedora-atomic/rawhide/x86_64/docker-host",
             ],
-            "add_arch_template": ["atomic-installer/lorax-embed-repo.tmpl"],
-            "add_arch_template_var": [
-                "ostree_repo=https://kojipkgs.fedoraproject.org/atomic/26/",
-                "ostree_osname=fedora-atomic",
-                "ostree_install_ref=fedora/26/x86_64/updates/atomic-host",
-                "ostree_update_ref=fedora/26/x86_64/atomic-host",
-            ]
             'template_repo': 'https://pagure.io/fedora-lorax-templates.git',
             'template_branch': 'f26',
         }