diff --git a/gpt_action_json_schema.json b/gpt_action_json_schema.json
index 0c5a44b..5f4e6ed 100644
--- a/gpt_action_json_schema.json
+++ b/gpt_action_json_schema.json
@@ -324,24 +324,6 @@
         ],
         "title": "ValidationError"
       },
-      "Writefile": {
-        "properties": {
-          "file_path": {
-            "type": "string",
-            "title": "File Path"
-          },
-          "file_content": {
-            "type": "string",
-            "title": "File Content"
-          }
-        },
-        "type": "object",
-        "required": [
-          "file_path",
-          "file_content"
-        ],
-        "title": "Writefile"
-      },
       "WritefileWithUUID": {
         "properties": {
           "file_path": {
diff --git a/tests/test_basic.py b/tests/test_basic.py
index 26c90e5..082bfbf 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -1,6 +1,6 @@
 import unittest
 from unittest.mock import patch, mock_open
-from src.wcgw.basic import text_from_editor, save_history, parse_user_message_special, Config
+from src.wcgw.client.basic import text_from_editor, save_history, parse_user_message_special, Config
 import os
 import tempfile
 import json
diff --git a/tests/test_tools.py b/tests/test_tools.py
index 6ff4d98..4ecc3d3 100644
--- a/tests/test_tools.py
+++ b/tests/test_tools.py
@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import patch
-from src.wcgw.tools import render_terminal_output, ask_confirmation, Confirmation
-from types_ import Writefile
+from src.wcgw.client.tools import render_terminal_output, ask_confirmation, Confirmation
+from src.wcgw.types_ import Writefile
 
 
 class TestTools(unittest.TestCase):