diff --git a/src/runtime/crt/host/microtvm_api_server.py b/src/runtime/crt/host/microtvm_api_server.py
index 7271c4512abc..031e4de1139f 100644
--- a/src/runtime/crt/host/microtvm_api_server.py
+++ b/src/runtime/crt/host/microtvm_api_server.py
@@ -218,7 +218,7 @@ def read_transport(self, n, timeout_sec):
             to_return = 0
 
         if not to_return:
-            self.disconnect_transport()
+            self.close_transport()
             raise server.TransportClosedError()
 
         return to_return
@@ -239,7 +239,7 @@ def write_transport(self, data, timeout_sec):
                 num_written = 0
 
             if not num_written:
-                self.disconnect_transport()
+                self.close_transport()
                 raise server.TransportClosedError()
 
             data = data[num_written:]