diff --git a/agent/tools/code_exec.py b/agent/tools/code_exec.py
index 1c5cae24d..be5b29561 100644
--- a/agent/tools/code_exec.py
+++ b/agent/tools/code_exec.py
@@ -17,7 +17,7 @@ import base64
 import logging
 import os
 from abc import ABC
-from enum import StrEnum
+from strenum import StrEnum
 from typing import Optional
 from pydantic import BaseModel, Field, field_validator
 from agent.tools.base import ToolParamBase, ToolBase, ToolMeta