diff --git a/HealthGPT/Onboarding/LLMSource.swift b/HealthGPT/Onboarding/LLMSource.swift
index 535b24b..0347059 100644
--- a/HealthGPT/Onboarding/LLMSource.swift
+++ b/HealthGPT/Onboarding/LLMSource.swift
@@ -7,7 +7,6 @@
 //
 
 import Foundation
-import SpeziLLMOpenAI
 
 
 enum LLMSource: String, CaseIterable, Identifiable, Codable {
diff --git a/HealthGPTUITests/OnboardingUITests.swift b/HealthGPTUITests/OnboardingUITests.swift
index 0d23c15..fd5a847 100644
--- a/HealthGPTUITests/OnboardingUITests.swift
+++ b/HealthGPTUITests/OnboardingUITests.swift
@@ -73,8 +73,16 @@ extension XCUIApplication {
         
         let picker = pickers["llmSourcePicker"]
         let optionToSelect = picker.pickerWheels.element(boundBy: 0)
-        optionToSelect.adjust(toPickerWheelValue: "Open AI LLM")
+        optionToSelect.adjust(toPickerWheelValue: "On-device LLM")
+        
+        XCTAssertTrue(buttons["Save Choice"].waitForExistence(timeout: 5))
+        buttons["Save Choice"].tap()
+        
+        XCTAssertTrue(staticTexts["LLM Download"].waitForExistence(timeout: 5))
+        XCTAssertTrue(buttons["Back"].waitForExistence(timeout: 2))
+        buttons["Back"].tap()
         
+        optionToSelect.adjust(toPickerWheelValue: "Open AI LLM")
         XCTAssertTrue(buttons["Save Choice"].waitForExistence(timeout: 5))
         buttons["Save Choice"].tap()
     }