diff --git a/cmd/pruner/pruner.go b/cmd/pruner/pruner.go
index abb540e..9655c95 100644
--- a/cmd/pruner/pruner.go
+++ b/cmd/pruner/pruner.go
@@ -12,7 +12,7 @@ import (
 )
 
 func main() {
-	db, err := util.GetDBConnection(os.Getenv("DB_ADDR"))
+	db, err := util.GetDBConnection(os.Getenv("DATABASE_ADDR"))
 	if err != nil {
 		log.Fatalln(err)
 	}
diff --git a/cmd/ws/ws.go b/cmd/ws/ws.go
index 23715ff..88b77c5 100644
--- a/cmd/ws/ws.go
+++ b/cmd/ws/ws.go
@@ -19,7 +19,7 @@ func main() {
 		log.Fatalln("WS_ADDR is not set.")
 	}
 
-	db, err := util.GetDBConnection(os.Getenv("DB_ADDR"))
+	db, err := util.GetDBConnection(os.Getenv("DATABASE_ADDR"))
 	if err != nil {
 		log.Fatalln(err)
 	}