diff --git a/tests/test_issue_2481.py b/tests/test_issue_2481.py
new file mode 100644
index 000000000..7ff962889
--- /dev/null
+++ b/tests/test_issue_2481.py
@@ -0,0 +1,14 @@
+from smolagents import CodeAgent, InferenceClientModel, Tool
+
+image_generation_tool = Tool.from_space(
+    "black-forest-labs/FLUX.1-schnell",
+    name="image_generator",
+    description="Generate an image from a prompt"
+)
+
+model = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
+
+agent = CodeAgent(
+    tools=[image_generation_tool],
+    model=model
+)