2 files changed, 5 insertions, 4 deletions
diff --git a/extension/react-app/src/components/dialogs/AddContextGroupDialog.tsx b/extension/react-app/src/components/dialogs/AddContextGroupDialog.tsx
index 9cd0a95e..a6cf151c 100644
--- a/extension/react-app/src/components/dialogs/AddContextGroupDialog.tsx
+++ b/extension/react-app/src/components/dialogs/AddContextGroupDialog.tsx
@@ -1,5 +1,5 @@
 import { useContext } from "react";
-import { Button, TextInput } from "..";
+import { Button, Input } from "..";
 import { GUIClientContext } from "../../App";
 import { useDispatch } from "react-redux";
 import {
@@ -27,7 +27,7 @@ function AddContextGroupDialog({
 
   return (
     <div className="p-4">
-      <TextInput
+      <Input
         defaultValue="My Context Group"
         type="text"
         ref={(input) => {
diff --git a/extension/react-app/src/components/dialogs/FTCDialog.tsx b/extension/react-app/src/components/dialogs/FTCDialog.tsx
index 3ea753bc..5fa2d4e6 100644
--- a/extension/react-app/src/components/dialogs/FTCDialog.tsx
+++ b/extension/react-app/src/components/dialogs/FTCDialog.tsx
@@ -1,6 +1,6 @@
 import React, { useContext } from "react";
 import styled from "styled-components";
-import { Button, TextInput } from "..";
+import { Button, Input } from "..";
 import { useNavigate } from "react-router-dom";
 import { GUIClientContext } from "../../App";
 import { useDispatch } from "react-redux";
@@ -37,7 +37,7 @@ function FTCDialog() {
         OpenAIFreeTrial object.
       </p>
 
-      <TextInput
+      <Input
         type="text"
         placeholder="Enter your OpenAI API key"
         value={apiKey}
@@ -46,6 +46,7 @@ function FTCDialog() {
       <GridDiv>
         <Button
           onClick={() => {
+            dispatch(setShowDialog(false));
             navigate("/models");
           }}
         >