diff --git a/code.sh b/code.sh
index de8d0ec..2173486 100755
--- a/code.sh
+++ b/code.sh
@@ -4,6 +4,7 @@ set -e
 shopt -s nullglob
 
 FIRST_RUN="${XDG_CONFIG_HOME}/flatpak-vscode-first-run"
+WAYLAND_OPTS=""
 
 function msg() {
   echo "flatpak-vscode: $*" >&2
@@ -14,6 +15,10 @@ if [ ! -f ${FIRST_RUN} ]; then
   touch ${FIRST_RUN}
 fi
 
+if [ "$XDG_SESSION_TYPE" = "wayland" ]; then
+  WAYLAND_OPTS="--ozone-platform-hint=auto"
+fi
+
 PYTHON_SITEDIR=$(python3 <<EOFPYTHON
 import os
 import site
@@ -67,4 +72,5 @@ fi
 exec env ELECTRON_RUN_AS_NODE=1 PATH="${PATH}:${XDG_DATA_HOME}/node_modules/bin" \
   /app/bin/zypak-wrapper.sh /app/extra/vscode/code /app/extra/vscode/resources/app/out/cli.js \
   --ms-enable-electron-run-as-node --extensions-dir=${XDG_DATA_HOME}/vscode/extensions \
+  ${WAYLAND_OPTS} \
   "$@" ${WARNING_FILE}
diff --git a/com.visualstudio.code.yaml b/com.visualstudio.code.yaml
index ab76c3f..90acce8 100644
--- a/com.visualstudio.code.yaml
+++ b/com.visualstudio.code.yaml
@@ -12,6 +12,7 @@ finish-args:
   - --require-version=0.10.3
   - --share=ipc
   - --socket=x11
+  - --socket=wayland
   - --socket=pulseaudio
   - --socket=ssh-auth
   - --share=network