diff --git a/update.sh b/update.sh
index 658d925..70764fc 100755
--- a/update.sh
+++ b/update.sh
@@ -1,6 +1,15 @@
 #!/bin/bash
 
+restart-bot() {
+    USERFILE=~/.config/systemd/user/discord-bot.service
+    if [ -f "$USERFILE" ]; then
+        systemctl --user restart discord-bot
+    else
+        systemctl restart discord-bot
+    fi
+}
+
 echo 'Updating...'
 git pull origin master
 echo 'Finished. Starting bot...'
-systemctl --user restart discord-bot
+restart-bot