diff --git a/src/providers/userTokenProvider/components/ModelSelection/index.tsx b/src/providers/userTokenProvider/components/ModelSelection/index.tsx
index 518aa27..0fab7a4 100644
--- a/src/providers/userTokenProvider/components/ModelSelection/index.tsx
+++ b/src/providers/userTokenProvider/components/ModelSelection/index.tsx
@@ -19,9 +19,7 @@ export function ModelSelection({ handleBack, handleClick, models }: Props) {
           </Button>
         ))}
         <div>
-          <Button variant="ghost" onClick={handleBack}>
-            Back
-          </Button>
+          <Button onClick={handleBack}>Back</Button>
         </div>
       </div>
     </div>
diff --git a/src/providers/userTokenProvider/index.tsx b/src/providers/userTokenProvider/index.tsx
index 97e01ca..72dd34f 100644
--- a/src/providers/userTokenProvider/index.tsx
+++ b/src/providers/userTokenProvider/index.tsx
@@ -346,6 +346,7 @@ export class UserTokenProvider extends BaseProvider {
     const [SelectedTaskComponent, setSelectedTaskComponent] = useState<React.ComponentType | null>(
       null,
     );
+    const [selectedTask, setSelectedTask] = useState("");
 
     useEffect(() => {
       if (isOpen && dialogRef.current) {
@@ -406,8 +407,11 @@ export class UserTokenProvider extends BaseProvider {
                   className={styles.taskButton}
                   key={task.name}
                   size="small"
-                  variant="primary"
-                  onClick={() => setSelectedTaskComponent(() => task.component)}
+                  variant={selectedTask === task.name ? "primary" : "ghost"}
+                  onClick={() => {
+                    setSelectedTaskComponent(() => task.component);
+                    setSelectedTask(task.name);
+                  }}
                 >
                   {task.name}
                 </Button>
@@ -525,10 +529,10 @@ export class UserTokenProvider extends BaseProvider {
           onChange={(e) => setInputValue(e.target.value)}
         />
         <div className={styles.buttonGroup}>
-          <Button type="submit">Submit</Button>
-          <Button variant="ghost" onClick={() => setSelectedModel(null)}>
-            Back
+          <Button type="submit" variant="accent">
+            Submit
           </Button>
+          <Button onClick={() => setSelectedModel(null)}>Back</Button>
         </div>
       </form>
     );
@@ -778,7 +782,7 @@ export class UserTokenProvider extends BaseProvider {
               }
             }}
           />
-          <Button disabled={isButtonDisabled} onClick={startTask}>
+          <Button disabled={isButtonDisabled} variant="accent" onClick={startTask}>
             {isButtonDisabled ? "Provide context to enable" : "Create Annotation"}
           </Button>
         </>
@@ -944,7 +948,9 @@ export class UserTokenProvider extends BaseProvider {
               setInputValue(currentTarget.value.trim());
             }}
           />
-          <Button onClick={startTask}>Start Transcription</Button>
+          <Button variant="accent" onClick={startTask}>
+            Start Transcription
+          </Button>
         </>
       );
     }