diff --git a/polymind/core/codegen.py b/polymind/core/codegen.py
index 29b48dd..684faee 100644
--- a/polymind/core/codegen.py
+++ b/polymind/core/codegen.py
@@ -10,7 +10,7 @@
 from abc import ABC, abstractmethod
 from typing import List
 
-from dspy import InputField, Module, OutputField, Signature
+# from dspy import InputField, Module, OutputField, Signature
 from pydantic import Field
 
 from polymind.core.logger import Logger