diff --git a/infra/docker/docker-compose.yml b/infra/docker/docker-compose.yml
index 1700f10..5eec49c 100644
--- a/infra/docker/docker-compose.yml
+++ b/infra/docker/docker-compose.yml
@@ -70,7 +70,7 @@ services:
     ports:
       - "127.0.0.1:8000:8000"
     healthcheck:
-      test: ["CMD", "wget", "--spider", "-q", "http://localhost:8000/health"]
+      test: ["CMD", "python", "-c", "import urllib.request,sys; sys.exit(0 if urllib.request.urlopen('http://localhost:8000/health',timeout=3).status==200 else 1)"]
       interval: 10s
       timeout: 5s
       retries: 5
@@ -219,7 +219,7 @@ services:
       retries: 5
 
   mlflow:
-    image: ghcr.io/mlflow/mlflow:2.14.3
+    image: ghcr.io/mlflow/mlflow:v2.14.3
     profiles: [mlops]
     command: >
       mlflow server
@@ -237,7 +237,7 @@ services:
     ports:
       - "127.0.0.1:5000:5000"
     healthcheck:
-      test: ["CMD", "curl", "--fail", "http://localhost:5000/health"]
+      test: ["CMD", "python", "-c", "import urllib.request,sys; sys.exit(0 if urllib.request.urlopen('http://localhost:5000/health',timeout=3).status==200 else 1)"]
       interval: 10s
       timeout: 5s
       retries: 5