diff --git a/requirements-swh.txt b/requirements-swh.txt
--- a/requirements-swh.txt
+++ b/requirements-swh.txt
@@ -2,5 +2,5 @@
 swh.core[db,http] >= 0.14
 swh.model >= 2.6.1
 swh.storage
-swh.graph
+swh.graph >= 2.0.0
 swh.journal
diff --git a/swh/provenance/swhgraph/archive.py b/swh/provenance/swhgraph/archive.py
--- a/swh/provenance/swhgraph/archive.py
+++ b/swh/provenance/swhgraph/archive.py
@@ -9,7 +9,7 @@
 import grpc
 
 from swh.core.statsd import statsd
-from swh.graph.rpc import swhgraph_pb2, swhgraph_pb2_grpc
+from swh.graph.grpc import swhgraph_pb2, swhgraph_pb2_grpc
 from swh.model.model import Sha1Git
 from swh.model.swhids import CoreSWHID, ObjectType
 from swh.storage.interface import StorageInterface
diff --git a/swh/provenance/tests/conftest.py b/swh/provenance/tests/conftest.py
--- a/swh/provenance/tests/conftest.py
+++ b/swh/provenance/tests/conftest.py
@@ -17,7 +17,7 @@
 import pytest
 from pytest_postgresql.factories import postgresql
 
-from swh.graph.http_server import make_app
+from swh.graph.http_rpc_server import make_app
 from swh.journal.serializers import msgpack_ext_hook
 from swh.model.model import BaseModel, TimestampWithTimezone
 from swh.provenance import get_provenance, get_provenance_storage