diff --git a/llama_toolchain/safety/client.py b/llama_toolchain/safety/client.py
index 3e03530c1..73c5682e3 100644
--- a/llama_toolchain/safety/client.py
+++ b/llama_toolchain/safety/client.py
@@ -5,6 +5,7 @@
 # the root directory of this source tree.
 
 import asyncio
+import json
 
 from typing import Any