diff --git a/stubs/google-cloud-ndb/google/cloud/ndb/context.pyi b/stubs/google-cloud-ndb/google/cloud/ndb/context.pyi
index 6889e9304bb9..d0c288628c39 100644
--- a/stubs/google-cloud-ndb/google/cloud/ndb/context.pyi
+++ b/stubs/google-cloud-ndb/google/cloud/ndb/context.pyi
@@ -86,6 +86,7 @@ class Context(_Context):
     def set_retry_state(self, state) -> None: ...
     def clear_retry_state(self) -> None: ...
     def call_on_commit(self, callback) -> None: ...
+    def call_on_transaction_complete(self, callback: Callable[[], object]) -> None: ...
     def in_transaction(self): ...
     def in_retry(self): ...
     def memcache_add(self, *args, **kwargs) -> None: ...