diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 1048260..bdaf2ad 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -1,5 +1,9 @@
 {
     "name": "Jupyter Notebook with MongoDB",
+    "features": {
+        "ghcr.io/devcontainers/features/python:1": {},
+        "ghcr.io/devcontainers/features/common-utils:2": {}
+    },
     "dockerComposeFile": "docker-compose.yml",
     "service": "lab-runner",
     "workspaceFolder": "/workspaces/${localWorkspaceFolderBasename}",
diff --git a/.devcontainer/docker-compose.yml b/.devcontainer/docker-compose.yml
index 2f8ffdf..700def2 100644
--- a/.devcontainer/docker-compose.yml
+++ b/.devcontainer/docker-compose.yml
@@ -1,6 +1,6 @@
 services:
   lab-runner:
-    image: mcr.microsoft.com/devcontainers/universal:2
+    image: mcr.microsoft.com/devcontainers/base:ubuntu
     volumes:
       - ../..:/workspaces:cached
     depends_on: