diff --git a/fish/conf.d/aliases.fish b/fish/conf.d/aliases.fish
index 89c2565..76c9e19 100644
--- a/fish/conf.d/aliases.fish
+++ b/fish/conf.d/aliases.fish
@@ -68,4 +68,5 @@ end
 
 # run code in wayland natively
 # TODO: just add it to the args file instead
-alias code="code --enable-features=UseOzonePlatform --ozone-platform=wayland"
+alias code="code --ozone-platform=wayland"
+alias vesktop="vesktop --ozone-platform=wayland"
diff --git a/restore.sh b/restore.sh
index d8a3405..e9bb9ae 100755
--- a/restore.sh
+++ b/restore.sh
@@ -129,9 +129,12 @@ elif command_exists "code-oss"; then
     cp $GIT_DIR/vscode/argv.json $HOME/.vscode-oss/argv.json
 fi
 # VS code: run in wayland natively
-sudo sed '/^Exec=\/usr\/share\/code\/code\( --new-window\)\? %F$/ s/$/ --enable-features=UseOzonePlatform --ozone-platform=wayland/' -i /usr/share/applications/code.desktop
+sudo sed '/^Exec=\/usr\/share\/code\/code\( --new-window\)\? %F$/ s/$/ --ozone-platform=wayland/' -i /usr/share/applications/code.desktop
 
-echo "restore.sh done!"
+# Vesktop: run in wayland natively
+sudo sed '/^Exec=\/opt\/Vesktop\/vesktop %U$/ s/$/ --ozone-platform=wayland/' -i /usr/share/applications/vesktop.desktop
 
 # default applications
 xdg-settings set default-web-browser librewolf.desktop
+
+echo "restore.sh done!"