diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
index 8a7c3d5..8692a29 100644
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -5,7 +5,6 @@ services:
     build:
       context: ./PostgreSQL
       dockerfile: Dockerfile
-    image: postgres
     restart: always
     environment:
       - POSTGRES_DB=${DB_NAME}
diff --git a/docker-compose.prod.yml b/docker-compose.prod.yml
index ed9bc76..a91d176 100644
--- a/docker-compose.prod.yml
+++ b/docker-compose.prod.yml
@@ -2,7 +2,9 @@ services:
   database:
     hostname: database
     container_name: briques_postgres
-    image: postgres
+    build:
+      context: ./PostgreSQL
+      dockerfile: Dockerfile
     restart: always
     environment:
       - POSTGRES_DB=${DB_NAME}