diff --git a/src/agents/model-compat.ts b/src/agents/model-compat.ts
index f4e39c8e9..c9a71bc09 100644
--- a/src/agents/model-compat.ts
+++ b/src/agents/model-compat.ts
@@ -1,4 +1,4 @@
-import type { Api, Model, OpenAICompletionsCompat } from "@mariozechner/pi-ai";
+import type { Api, Model, OpenAICompat } from "@mariozechner/pi-ai";
 
 function isOpenAiCompletionsModel(model: Model<Api>): model is Model<"openai-completions"> {
   return model.api === "openai-completions";
@@ -10,7 +10,7 @@ export function normalizeModelCompat(model: Model<Api>): Model<Api> {
   if (!isZai || !isOpenAiCompletionsModel(model)) return model;
 
   const openaiModel = model as Model<"openai-completions">;
-  const compat = openaiModel.compat as OpenAICompletionsCompat | undefined;
+  const compat = openaiModel.compat as OpenAICompat | undefined;
   if (compat?.supportsDeveloperRole === false) return model;
 
   openaiModel.compat = compat
diff --git a/src/tui/components/custom-editor.ts b/src/tui/components/custom-editor.ts
index 50f810c1c..c970cb2a6 100644
--- a/src/tui/components/custom-editor.ts
+++ b/src/tui/components/custom-editor.ts
@@ -1,10 +1,8 @@
 import {
   Editor,
-  type EditorOptions,
   type EditorTheme,
   Key,
   matchesKey,
-  type TUI,
 } from "@mariozechner/pi-tui";
 
 export class CustomEditor extends Editor {
@@ -19,8 +17,8 @@ export class CustomEditor extends Editor {
   onShiftTab?: () => void;
   onAltEnter?: () => void;
 
-  constructor(tui: TUI, theme: EditorTheme, options?: EditorOptions) {
-    super(tui, theme, options);
+  constructor(theme: EditorTheme) {
+    super(theme);
   }
   handleInput(data: string): void {
     if (matchesKey(data, Key.alt("enter")) && this.onAltEnter) {
diff --git a/src/tui/tui.ts b/src/tui/tui.ts
index a5e6e34d7..753f5511f 100644
--- a/src/tui/tui.ts
+++ b/src/tui/tui.ts
@@ -193,7 +193,7 @@ export async function runTui(opts: TuiOptions) {
   const statusContainer = new Container();
   const footer = new Text("", 1, 0);
   const chatLog = new ChatLog();
-  const editor = new CustomEditor(tui, editorTheme);
+  const editor = new CustomEditor(editorTheme);
   const root = new Container();
   root.addChild(header);
   root.addChild(chatLog);