diff --git a/python/orca/src/bigdl/orca/learn/pytorch/pytorch_pyspark_estimator.py b/python/orca/src/bigdl/orca/learn/pytorch/pytorch_pyspark_estimator.py
index b7bc604ee83..63d3527f2bb 100644
--- a/python/orca/src/bigdl/orca/learn/pytorch/pytorch_pyspark_estimator.py
+++ b/python/orca/src/bigdl/orca/learn/pytorch/pytorch_pyspark_estimator.py
@@ -569,4 +569,4 @@ def shutdown(self):
         Shutdown estimator and release resources.
         """
         if self.need_to_log_to_driver:
-            stop_log_server(self.log_server_thread, self.ip, self.port)
+            stop_log_server(self.log_server_thread, self.ip, self.log_port)