diff --git a/examples/storage/parttool/pytest_parttool_example.py b/examples/storage/parttool/pytest_parttool_example.py
index 82dcebd29c..0f4cfc5f11 100644
--- a/examples/storage/parttool/pytest_parttool_example.py
+++ b/examples/storage/parttool/pytest_parttool_example.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: 2022-2023 Espressif Systems (Shanghai) CO LTD
+# SPDX-FileCopyrightText: 2022-2024 Espressif Systems (Shanghai) CO LTD
 # SPDX-License-Identifier: Unlicense OR CC0-1.0
 import os
 import subprocess
@@ -16,8 +16,7 @@ def test_examples_parttool(dut: Dut) -> None:
     dut.expect('Example end')
 
     # Close connection to DUT
-    dut.pexpect_proc.terminate()
-    dut.serial.stop_redirect_thread()
+    dut.serial.close()
 
     # Run the example python script
     idf_path = os.getenv('IDF_PATH')