diff --git a/config/default.go b/config/default.go
index 0fc94afa..90e2addd 100644
--- a/config/default.go
+++ b/config/default.go
@@ -55,7 +55,8 @@ func NewConfig() *Config {
 			Datasources: []string{"configdrive:/media/config-2"},
 		},
 		Upgrade: UpgradeConfig{
-			Url: "http://storage.googleapis.com/releases_rancher_com/rancheros/versions.yml",
+			Url:   "https://releases.rancher.com/os/versions.yml",
+			Image: "rancher/os",
 		},
 		BootstrapContainers: []ContainerConfig{
 			{
diff --git a/config/types.go b/config/types.go
index d0376abf..aa52cb49 100644
--- a/config/types.go
+++ b/config/types.go
@@ -26,6 +26,7 @@ type ContainerConfig struct {
 	Cmd            string `yaml:"run,omitempty"`
 	MigrateVolumes bool   `yaml:"migrate_volumes,omitempty"`
 	ReloadConfig   bool   `yaml:"reload_config,omitempty"`
+	CreateOnly     bool   `yaml:create_only,omitempty`
 }
 
 type Config struct {
@@ -53,7 +54,9 @@ type ConsoleConfig struct {
 }
 
 type UpgradeConfig struct {
-	Url string `yaml:"url,omitempty"`
+	Url      string `yaml:"url,omitempty"`
+	Image    string `yaml:"image,omitempty"`
+	Rollback string `yaml:"rollback,omitempty"`
 }
 
 type DnsConfig struct {