diff --git a/swh/graph/cli.py b/swh/graph/cli.py
index edf285c5e59dca990710528c496aa98bd25337a0..8afd692701343bdd24da4c47734c1cfd2dfca50b 100644
--- a/swh/graph/cli.py
+++ b/swh/graph/cli.py
@@ -175,6 +175,7 @@ def grpc_serve(ctx, port, java_home, graph):
     config = ctx.obj["config"]
     config.setdefault("graph", {})
     config["graph"]["path"] = graph
+    config["graph"]["port"] = port
 
     logger.debug("Building gPRC server command line")
     cmd, port = build_grpc_server_cmdline(**config["graph"])