diff --git a/ct/flame.sh b/ct/flame.sh
index 7e4aeeac..ba2b2227 100644
--- a/ct/flame.sh
+++ b/ct/flame.sh
@@ -44,6 +44,7 @@ function update_script() {
     msg_info "Restoring Data"
     cp -r /opt/flame_data_backup/. /opt/flame/data
     cp /opt/flame.env.bak /opt/flame/.env
+    sed -i "s/^VERSION=.*/VERSION=$(cat ~/.flame)/" /opt/flame/.env
     rm -rf /opt/flame_data_backup /opt/flame.env.bak
     msg_ok "Restored Data"
 
diff --git a/install/flame-install.sh b/install/flame-install.sh
index 8f592243..54e7fab5 100644
--- a/install/flame-install.sh
+++ b/install/flame-install.sh
@@ -32,8 +32,10 @@ $STD npm install --production
 $STD npm run build
 cd /opt/flame
 cp -r client/build/. public/
+FLAME_VERSION=$(cat ~/.flame)
 cat <<EOF >/opt/flame/.env
 NODE_ENV=production
+VERSION=${FLAME_VERSION}
 PASSWORD=
 EOF
 msg_ok "Set up Flame"