diff --git a/code-server/docker-compose.yml b/code-server/docker-compose.yml
index 3ac012b..9285e3c 100644
--- a/code-server/docker-compose.yml
+++ b/code-server/docker-compose.yml
@@ -3,8 +3,12 @@ services:
     environment:
       PUID: "${HOST_UID}"
       PGID: "${HOST_GID}"
+      UMASK: "${HOST_UMASK:-022}"
+      # See https://mods.linuxserver.io/?mod=code-server for the list of MODs available
+      DOCKER_MODS: "linuxserver/mods:universal-docker"
     volumes:
       - /var/run/docker.sock:/var/run/docker.sock
       - /:/mnt/runtipi.host
     labels:
-      traefik.http.routers.code-server.middlewares: authentik_sysmgr@file
\ No newline at end of file
+      traefik.http.routers.code-server.middlewares: authentik_sysmgr@file
+      
\ No newline at end of file