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