diff --git a/src/codexmcp/server.py b/src/codexmcp/server.py
index cb79b47..168d758 100644
--- a/src/codexmcp/server.py
+++ b/src/codexmcp/server.py
@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-from ast import List
 import json
 import os
 import queue
@@ -10,7 +9,7 @@ import subprocess
 import threading
 import uuid
 from pathlib import Path
-from typing import Annotated, Any, Dict, Generator, Literal, Optional
+from typing import Annotated, Any, Dict, Generator, List, Literal, Optional
 
 from mcp.server.fastmcp import FastMCP
 from pydantic import BeforeValidator, Field
@@ -149,12 +148,12 @@ async def codex(
         bool,
         "Return all messages (e.g. reasoning, tool calls, etc.) from the codex session. Set to `False` by default, only the agent's final reply message is returned.",
     ] = False,
-    # image: Annotated[
-    #     Optional[List[Path]],
-    #     Field(
-    #         description="Attach one or more image files to the initial prompt. Separate multiple paths with commas or repeat the flag.",
-    #     ),
-    # ] = None,
+    image: Annotated[
+        Optional[List[Path]],
+        Field(
+            description="Attach one or more image files to the initial prompt. Separate multiple paths with commas or repeat the flag.",
+        ),
+    ] = None,
     model: Annotated[
         Optional[str],
         Field(
@@ -176,8 +175,8 @@ async def codex(
     # Build command as list to avoid injection
     cmd = ["codex", "exec", "--sandbox", sandbox, "--cd", str(cd), "--json"]
     
-    # if image is not None:
-    #     cmd.extend(["--image", ",".join(image)])
+    if image is not None:
+        cmd.extend(["--image", ",".join(image)])
         
     if model is not None:
         cmd.extend(["--model", model])