diff --git a/src/hive_cli/libs/common_tools.py b/src/hive_cli/libs/common_tools.py
index 9628f69..b404ad1 100644
--- a/src/hive_cli/libs/common_tools.py
+++ b/src/hive_cli/libs/common_tools.py
@@ -3,10 +3,8 @@
 import io
 import signal
 import subprocess
-import threading
 import time
 
-import psutil
 import requests
 
 GCR_SANDBOX_BUCKET = "hi-sandbox"
@@ -55,7 +53,7 @@ def run_command(
         raise FunctionExecutionError(error_code_to_string(-process.returncode))
       if process.returncode != 0:
         raise FunctionExecutionError(f"Error: {stderr}")
-      return stdout
+      return stdout.strip().splitlines()[-1]  # Return only the last line of output
     except subprocess.TimeoutExpired as exc:
       process.kill()
       raise FunctionExecutionError("Timeout") from exc