diff --git a/gnome-config.sh b/gnome-config.sh
index 401f8be..9420893 100644
--- a/gnome-config.sh
+++ b/gnome-config.sh
@@ -9,5 +9,21 @@ gsettings set org.gnome.settings-daemon.plugins.xsettings overrides "{'Gtk/Butto
 # To reduce GNOME 3 titlebar height
 cp gtk.css ~/.config/gtk-3.0/gtk.css
 
-# Restart gnome shell - manually with  alt+f2, type r and press return (not in budgie)
-gnome-shell --replace
\ No newline at end of file
+
+# Restart gnome shell if its running - manually with  alt+f2, type restart and press return (not in budgie)
+if pgrep -x "gedit" > /dev/null
+then
+    echo "Restarting Gnome shell"
+    gnome-shell --replace
+else
+    echo "Gnome shell is not running so not restarting it"
+fi
+
+# If budgie is running restart budgie
+if pgrep -x "gedit" > /dev/null
+then
+    echo "Restarting Budgie window manager"
+    budgie-wm --replace
+else
+    echo "Budgie window manager is not running so not restarting it"
+fi
\ No newline at end of file