diff --git a/api/__init__.py b/api/__init__.py
index 00c1638..c24d109 100644
--- a/api/__init__.py
+++ b/api/__init__.py
@@ -16,7 +16,7 @@ from .database import update_snapshot, update_snapshot_for_current_operation
 from .database import delete_snapshot, delete_snapshot_by_operation
 from .database import get_operation_by_snapshot, get_snapshot_by_operation
 from .database import pick_snapshot
-from .database import pick_operation, sync_with_server, get_restore_operation, set_restore_operation
+from .database import pick_operation, sync_with_server, get_restore_operation, set_restore_operation, set_restore_operation_to_current
 
 from .batch_cmd import execute_batch_command
 from .batch_cmds import execute_batch_commands
diff --git a/api/database.py b/api/database.py
index e82a702..6d9089f 100644
--- a/api/database.py
+++ b/api/database.py
@@ -335,3 +335,7 @@ def get_restore_operation(name: str) -> int:
 
 def set_restore_operation(name: str, operation: int) -> None:
     write(name, f'update restore_operation set id = {operation}')
+
+
+def set_restore_operation_to_current(name: str) -> None:
+    return set_restore_operation(name, get_current_operation(name))
diff --git a/tjnetwork.py b/tjnetwork.py
index 12b9247..12bfeb4 100644
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -288,6 +288,9 @@ def get_restore_operation(name: str) -> int:
 def set_restore_operation(name: str, operation: int) -> None:
     return api.set_restore_operation(name, operation)
 
+def set_restore_operation_to_current(name: str) -> None:
+    return api.set_restore_operation_to_current(name)
+
 
 ############################################################
 # type