diff --git a/llmage/utils.py b/llmage/utils.py
index f4c393f..e3dde1c 100644
--- a/llmage/utils.py
+++ b/llmage/utils.py
@@ -1,6 +1,7 @@
 import json
 import time
 import asyncio
+import aiofiles
 from random import randint
 from functools import partial
 from traceback import format_exc