diff --git a/sway/restore_display.sh b/sway/restore_display.sh
index 64dd332..8c43fe6 100755
--- a/sway/restore_display.sh
+++ b/sway/restore_display.sh
@@ -4,6 +4,6 @@ OLDPWD=$OLDPWD
 PWD=$PWD
 cd $HOME/scripts/sway/
 source ~/.dotfiles/bash/plenv-path.sh
-perl -Ilocal/lib/perl5/5.38.2 displays.pl
+perl -Ilocal/lib/perl5/5.38.2 displays.pl $@
 cd $OLDPWD
 cd $PWD
diff --git a/tofi/power/↻ Reload Toolbar b/tofi/power/↻ Reload Toolbar
index 100df7a..c22dc85 100755
--- a/tofi/power/↻ Reload Toolbar	
+++ b/tofi/power/↻ Reload Toolbar	
@@ -2,6 +2,5 @@
 if [ -z ${SWAYSOCK+x} ]; then
     i3 reload
 else
-    cd ${HOME}/scripts/sway
-    perl -Ilocal/lib/perl5/5.38.2 display.pl -w
+    ${HOME}/scripts/sway/restore_display.sh -w 
 fi