diff --git a/autogen/code_utils.py b/autogen/code_utils.py
index ec801ccc9f38..d59a283bedbc 100644
--- a/autogen/code_utils.py
+++ b/autogen/code_utils.py
@@ -10,8 +10,6 @@
 from hashlib import md5
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-from _pytest import python
-
 from autogen import oai
 
 import docker
diff --git a/autogen/coding/local_commandline_code_executor.py b/autogen/coding/local_commandline_code_executor.py
index d6588a582054..fc3f4d5b29fb 100644
--- a/autogen/coding/local_commandline_code_executor.py
+++ b/autogen/coding/local_commandline_code_executor.py
@@ -144,7 +144,7 @@ def execute_code_blocks(self, code_blocks: List[CodeBlock]) -> CommandLineCodeRe
                 f.write(code)
             file_names.append(written_file)
 
-            program = sys.executable if _cmd(lang).startswith("python") else _cmd(lang)
+            program = sys.executable if lang.startswith("python") else _cmd(lang)
             cmd = [program, str(written_file.absolute())]
 
             try: