diff --git a/substra/sdk/backends/base.py b/substra/sdk/backends/base.py
index 91e35560..9a29d09d 100644
--- a/substra/sdk/backends/base.py
+++ b/substra/sdk/backends/base.py
@@ -1,4 +1,5 @@
 import abc
+from typing import List
 
 from substra.sdk.schemas import BackendType
 
@@ -34,7 +35,7 @@ def add_compute_plan_tuples(self, spec, spec_options):
         raise NotImplementedError
 
     @abc.abstractmethod
-    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys):
+    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys) -> List[str]:
         raise NotImplementedError
 
     @abc.abstractmethod
diff --git a/substra/sdk/backends/local/backend.py b/substra/sdk/backends/local/backend.py
index 2cd02521..c07b8149 100644
--- a/substra/sdk/backends/local/backend.py
+++ b/substra/sdk/backends/local/backend.py
@@ -750,7 +750,7 @@ def add(self, spec, spec_options=None, key=None):
                 add_asset(key, spec, spec_options)
                 return key
 
-    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys):
+    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys) -> List[str]:
         dataset = self._db.get(schemas.Type.Dataset, dataset_key)
         data_samples = list()
         for key in data_sample_keys:
diff --git a/substra/sdk/backends/remote/backend.py b/substra/sdk/backends/remote/backend.py
index 57fba465..e305e4ae 100644
--- a/substra/sdk/backends/remote/backend.py
+++ b/substra/sdk/backends/remote/backend.py
@@ -266,7 +266,7 @@ def add_compute_plan_tuples(self, spec, spec_options):
             key=spec.key,
         )
 
-    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys):
+    def link_dataset_with_data_samples(self, dataset_key, data_sample_keys) -> List[str]:
         """Returns the list of the data sample keys"""
         data = {
             "data_manager_keys": [dataset_key],
@@ -278,6 +278,7 @@ def link_dataset_with_data_samples(self, dataset_key, data_sample_keys):
             path="bulk_update/",
             data=data,
         )
+        return data_sample_keys
 
     def _download(self, url: str, destination_file: str) -> str:
         response = self._client.get_data(url, stream=True)