1 files changed, 3 insertions, 3 deletions
diff --git a/tools/run_tests/xds_k8s_test_driver/framework/test_app/runners/k8s/k8s_xds_server_runner.py b/tools/run_tests/xds_k8s_test_driver/framework/test_app/runners/k8s/k8s_xds_server_runner.py
index 20d3186c9c..b6de2f85e0 100644
--- a/tools/run_tests/xds_k8s_test_driver/framework/test_app/runners/k8s/k8s_xds_server_runner.py
+++ b/tools/run_tests/xds_k8s_test_driver/framework/test_app/runners/k8s/k8s_xds_server_runner.py
@@ -14,7 +14,6 @@
 """
 Run xDS Test Client on Kubernetes.
 """
-import datetime
 import logging
 from typing import List, Optional
 
@@ -197,7 +196,7 @@ class KubernetesServerRunner(k8s_base_runner.KubernetesBaseRunner):
         # Verify the deployment reports all pods started as well.
         self._wait_deployment_with_available_replicas(self.deployment_name,
                                                       replica_count)
-        self.time_start_completed = datetime.datetime.now()
+        self._start_completed()
 
         servers: List[XdsTestServer] = []
         for pod in pods:
@@ -239,6 +238,7 @@ class KubernetesServerRunner(k8s_base_runner.KubernetesBaseRunner):
 
     # pylint: disable=arguments-differ
     def cleanup(self, *, force=False, force_namespace=False):
+        # TODO(sergiitk): rename to stop().
         try:
             if self.deployment or force:
                 self._delete_deployment(self.deployment_name)
@@ -256,7 +256,7 @@ class KubernetesServerRunner(k8s_base_runner.KubernetesBaseRunner):
                 self.service_account = None
             self._cleanup_namespace(force=(force_namespace and force))
         finally:
-            self.time_stopped = datetime.datetime.now()
+            self._stop()
 
     # pylint: enable=arguments-differ