diff --git a/SignalServiceKit/tests/Messages/OWSLinkPreviewTest.swift b/SignalServiceKit/tests/Messages/OWSLinkPreviewTest.swift
index 79d93e0d0..f6e0c86e5 100644
--- a/SignalServiceKit/tests/Messages/OWSLinkPreviewTest.swift
+++ b/SignalServiceKit/tests/Messages/OWSLinkPreviewTest.swift
@@ -213,11 +213,11 @@ class OWSLinkPreviewTest: SSKBaseTestSwift {
         let expectation = self.expectation(description: "link download and parsing")
 
         OWSLinkPreview.tryToBuildPreviewInfo(previewUrl: "https://www.youtube.com/watch?v=tP-Ipsat90c")
-            .done { (draft) in
+            .done { (draft: OWSLinkPreviewDraft) in
                 XCTAssertNotNil(draft)
 
                 XCTAssertEqual(draft.title, "Randomness is Random - Numberphile")
-                XCTAssertNotNil(draft.imageFilePath)
+                XCTAssertNotNil(draft.jpegImageData)
 
                 expectation.fulfill()
             }.catch { (error) in