diff --git a/mlx-swift b/mlx-swift
index 1a2b3af..34bc52f 160000
--- a/mlx-swift
+++ b/mlx-swift
@@ -1 +1 @@
-Subproject commit 1a2b3af5f6c9b96147f6eaf8e75c7d24cfa2f377
+Subproject commit 34bc52fa1efd5a3bb6cf07b5de803f0cf7cdd3ab
diff --git a/mlx-swift-lm b/mlx-swift-lm
index 8055044..4e6e521 160000
--- a/mlx-swift-lm
+++ b/mlx-swift-lm
@@ -1 +1 @@
-Subproject commit 8055044c045aa879b5c420e7371d0896df92534b
+Subproject commit 4e6e5216852bfa9db031ab7c59cef801b6baa356
diff --git a/tests/SwiftBuddyTests/VLMExtractionTests.swift b/tests/SwiftBuddyTests/VLMExtractionTests.swift
index d89246a..e4b2a72 100644
--- a/tests/SwiftBuddyTests/VLMExtractionTests.swift
+++ b/tests/SwiftBuddyTests/VLMExtractionTests.swift
@@ -21,7 +21,7 @@ final class VLMExtractionTests: XCTestCase {
         let images = message.extractImages()
         XCTAssertEqual(images.count, 1)
         
-        if case let .ciImage(image) = images.first {
+        if case .ciImage(let image)? = images.first?.source {
             XCTAssertNotNil(image)
             XCTAssertEqual(image.extent.width, 1)
             XCTAssertEqual(image.extent.height, 1)
@@ -44,7 +44,7 @@ final class VLMExtractionTests: XCTestCase {
         let images = message.extractImages()
         XCTAssertEqual(images.count, 1)
         
-        if case let .url(url) = images.first {
+        if case .url(let url)? = images.first?.source {
             XCTAssertEqual(url.absoluteString, "https://example.com/test.jpg")
         } else {
             XCTFail("Expected .url, got \(String(describing: images.first))")