diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 204078f2..4ac066f5 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -27,7 +27,7 @@ repos:
         args: [ --config-file, pyproject.toml ]
         pass_filenames: false
         additional_dependencies:
-          - neptune-api==0.3.0
+          - neptune-api==0.4.0
           - more-itertools
 default_language_version:
     python: python3
diff --git a/pyproject.toml b/pyproject.toml
index a9831192..f5fab8f8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,7 +11,7 @@ pattern = "default-unprefixed"
 [tool.poetry.dependencies]
 python = "^3.8"
 
-neptune-api = "0.3.0"
+neptune-api = "0.4.0"
 more-itertools = "^10.0.0"
 
 [tool.poetry]
diff --git a/src/neptune_scale/__init__.py b/src/neptune_scale/__init__.py
index ed80271f..ec1a4127 100644
--- a/src/neptune_scale/__init__.py
+++ b/src/neptune_scale/__init__.py
@@ -197,7 +197,7 @@ def _create_run(
                 creation_time=None if creation_time is None else datetime_to_proto(creation_time),
             ),
         )
-        self._backend.submit(operation=operation)
+        self._backend.submit(operation=operation, family=self._family)
         # TODO: Enqueue on the operations queue
         # self._operations_queue.enqueue(operation=operation)
 
@@ -265,6 +265,6 @@ def log(
         )
 
         for operation in splitter:
-            self._backend.submit(operation=operation)
+            self._backend.submit(operation=operation, family=self._family)
             # TODO: Enqueue on the operations queue
             # self._operations_queue.enqueue(operation=operation)
diff --git a/src/neptune_scale/api/api_client.py b/src/neptune_scale/api/api_client.py
index b8b68369..80a15d31 100644
--- a/src/neptune_scale/api/api_client.py
+++ b/src/neptune_scale/api/api_client.py
@@ -43,8 +43,8 @@ def __init__(self, api_token: str) -> None:
         config, token_urls = get_config_and_token_urls(credentials=credentials)
         self._backend = create_auth_api_client(credentials=credentials, config=config, token_refreshing_urls=token_urls)
 
-    def submit(self, operation: RunOperation) -> None:
-        _ = submit_operation.sync(client=self._backend, body=operation)
+    def submit(self, operation: RunOperation, family: str) -> None:
+        _ = submit_operation.sync(client=self._backend, family=family, body=operation)
 
     def cleanup(self) -> None:
         pass