diff --git a/learnware/client/container.py b/learnware/client/container.py
index 81d7a5f..ed2df2a 100644
--- a/learnware/client/container.py
+++ b/learnware/client/container.py
@@ -113,7 +113,7 @@ class ModelCondaContainer(ModelContainer):
 
         input_shape = output_results["metadata"]["input_shape"]
         output_shape = output_results["metadata"]["output_shape"]
-        print("input_shape", input_shape, "output_shape", output_shape)
+        logger.info(f"input_shape: {input_shape}, output_shape: {output_shape}")
         self.reset(input_shape=input_shape, output_shape=output_shape)
 
     def _run_model_with_script(self, method, **kargs):
@@ -206,9 +206,8 @@ class ModelDockerContainer(ModelContainer):
 
     @staticmethod
     def _destroy_docker_container(docker_container):
-        # docker_container.stop()
-        # docker_container.remove()
-        pass
+        docker_container.stop()
+        docker_container.remove()
 
     def _copy_file_to_container(self, local_path, container_path):
         directory_path = os.path.dirname(container_path)
@@ -395,7 +394,7 @@ class ModelDockerContainer(ModelContainer):
 
         input_shape = output_results["metadata"]["input_shape"]
         output_shape = output_results["metadata"]["output_shape"]
-        print("input_shape", input_shape, "output_shape", output_shape)
+        logger.info(f"input_shape: {input_shape}, output_shape: {output_shape}")
         self.reset(input_shape=input_shape, output_shape=output_shape)
 
     def _init_env(self):
@@ -575,5 +574,4 @@ class LearnwaresContainer:
         learnware_containers = [
             _learnware for _learnware, _result in zip(self.learnware_containers, self.results) if _result is True
         ]
-        print("233", learnware_containers, list(self.results))
         return learnware_containers