diff --git a/README.md b/README.md
index b54d2cf..318937f 100644
--- a/README.md
+++ b/README.md
@@ -37,3 +37,4 @@ TODO:
 - figure out how to script firefox/librewolf extension installation
   - add vencord firefox
 - maybe just rewrite it in rust or something entirely??? idk
+- fix swaylock background generation
diff --git a/fish/conf.d/aliases.fish b/fish/conf.d/aliases.fish
index 0b3510b..b147d35 100644
--- a/fish/conf.d/aliases.fish
+++ b/fish/conf.d/aliases.fish
@@ -18,7 +18,7 @@ alias pls="please"
 
 if status is-interactive
     # kitty stuff
-    if [ $TERM = "xterm-kitty" ]
+    if [ $TERM = xterm-kitty ]
         alias icat="kitten icat"
         alias s="kitten ssh"
     else
@@ -26,7 +26,7 @@ if status is-interactive
         alias s="ssh"
     end
 
-# NixOS
+    # NixOS
     if type -q nixos-rebuild
         alias nrs="sudo NIXPKGS_ALLOW_INSECURE=1 nixos-rebuild switch --upgrade-all"
         alias nrb="sudo NIXPKGS_ALLOW_INSECURE=1 nixos-rebuild build --upgrade-all"
@@ -62,3 +62,7 @@ alias venv="if ! test -d ./.venv/; python3 -m venv .venv; end; source ./.venv/bi
 alias py-dep="pip install -r requirements.txt"
 
 alias sway="sway --unsupported-gpu"
+
+# run code in wayland natively
+# TODO: just add it to the args file instead
+alias code="code --enable-features=UseOzonePlatform --ozone-platform=wayland"
diff --git a/restore.sh b/restore.sh
index 33bf655..d8f0f56 100755
--- a/restore.sh
+++ b/restore.sh
@@ -126,6 +126,8 @@ elif command_exists "code-oss"; then
     code-oss $(for ext in $(cat $GIT_DIR/vscode/extensions.txt); do echo -n "--install-extension $ext "; done)
     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
 
 echo "restore.sh done!"