diff --git a/build/Dockerfile b/build/Dockerfile
index c86370d..b856e8d 100644
--- a/build/Dockerfile
+++ b/build/Dockerfile
@@ -1,5 +1,3 @@
-#ARG UPSTREAM_VERSION
-#FROM graphprotocol/graph-node:${UPSTREAM_VERSION}
-
-FROM graphprotocol/graph-node:latest
+ARG UPSTREAM_VERSION=latest
+FROM graphprotocol/graph-node:${UPSTREAM_VERSION}
 
diff --git a/dappnode_package.json b/dappnode_package.json
index 72ed3bf..ab1d421 100644
--- a/dappnode_package.json
+++ b/dappnode_package.json
@@ -2,7 +2,13 @@
   "name": "graph.dnp.dappnode.eth",
   "version": "0.1.0",
   "description": "The Graph is an indexing protocol for querying networks like Ethereum and IPFS. Anyone can build and publish open APIs, called subgraphs, making data easily accessible.",
-  "upstreamRepo": "graphprotocol/graph-node",
+  "upstream": [
+    {
+      "repo": "graphprotocol/graph-node",
+      "version": "latest",
+      "arg": "UPSTREAM_VERSION"
+    }
+  ],
   "type": "service",
   "author": "DAppNode Association <admin@dappnode.io> (https://github.com/dappnode)",
   "contributors": ["pablo <pablo@dappnode.io> (https://github.com/pablomendezroyo)"],
diff --git a/docker-compose.yml b/docker-compose.yml
index 8ebe919..e089e7c 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,6 +11,8 @@ services:
     image: "graph.dnp.dappnode.eth.graph.public.dappnode.eth:0.1.0"
     build:
       context: build
+      args:
+        UPSTREAM_VERSION: latest
     ports:
       - "8000:8000"
       - "8001:8001"