diff --git a/instructor/distil.py b/instructor/distil.py
index 894a80384..614652797 100644
--- a/instructor/distil.py
+++ b/instructor/distil.py
@@ -10,6 +10,8 @@
 import uuid
 import openai
 
+from instructor.function_calls import openai_schema
+
 
 class FinetuneFormat(enum.Enum):
     MESSAGES: str = "messages"