diff --git a/scripts/upgrade b/scripts/upgrade
index 771b98b..b15d9af 100644
--- a/scripts/upgrade
+++ b/scripts/upgrade
@@ -32,18 +32,19 @@ sudo sed -i "s@__PATH__@$app@g" ../conf/nginx.conf*
 sudo sed -i "s@__FINALPATH__@$final_path/@g" ../conf/nginx.conf*
 
 # Stop Mastodon Services
-sudo systemctl stop /etc/systemd/system/mastodon-*.service
+sudo systemctl stop mastodon-*
 
 # Update Mastodon
 sudo su - $app <<COMMANDS
 pushd ~/live
 git pull
+bundle install
 RAILS_ENV=production bin/bundle exec rails db:migrate
 RAILS_ENV=production bin/bundle exec rails assets:precompile
 COMMANDS
 
 # Start Mastodon Services
-sudo systemctl start /etc/systemd/system/mastodon-*.service
+sudo systemctl start mastodon-*
 
 # If app is public, add url to SSOWat conf as skipped_uris
 if [ $is_public = "Yes" ];