diff --git a/docker-compose.dev.yaml b/docker-compose.dev.yaml
index f6570a5..31493bf 100644
--- a/docker-compose.dev.yaml
+++ b/docker-compose.dev.yaml
@@ -1,7 +1,7 @@
 services:
   database:
-    image: postgres:17.2
-    restart: on-failure
+    image: postgres:17
+    restart: always
     volumes:
       - database:/var/lib/postgresql/data
     environment:
diff --git a/docker-compose.yaml b/docker-compose.yaml
index eba1e94..57a423f 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -1,7 +1,7 @@
 services:
   server:
     build: .
-    restart: on-failure
+    restart: unless-stopped
     depends_on:
       - database
     user: ${CONTAINER_UID:-0}:${CONTAINER_GID:-0}
@@ -27,8 +27,8 @@ services:
       - ${PORT:-80}:3000
 
   database:
-    image: postgres:17.2-alpine
-    restart: on-failure
+    image: postgres:17-alpine
+    restart: unless-stopped
     user: ${CONTAINER_UID:-0}:${CONTAINER_GID:-0}
     volumes:
       - ./data/database:/var/lib/postgresql/data
diff --git a/package.json b/package.json
index b4eb70b..7cee967 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "arkvault",
   "private": true,
-  "version": "0.4.0",
+  "version": "0.5.0",
   "type": "module",
   "scripts": {
     "dev": "vite dev",