1 files changed, 6 insertions, 3 deletions
diff --git a/Hotline/macOS/Files/FilePreviewTextView.swift b/Hotline/macOS/Files/FilePreviewTextView.swift
index c286381..4e3a719 100644
--- a/Hotline/macOS/Files/FilePreviewTextView.swift
+++ b/Hotline/macOS/Files/FilePreviewTextView.swift
@@ -11,7 +11,7 @@ struct FilePreviewTextView: View {
   @Environment(\.dismiss) var dismiss
   
   @Binding var info: PreviewFileInfo?
-  @State var preview: FilePreview? = nil
+  @State var preview: FilePreviewState? = nil
   @FocusState private var focusField: FilePreviewFocus?
   
   var body: some View {
@@ -89,7 +89,10 @@ struct FilePreviewTextView: View {
         if let info = info {
           ToolbarItem(placement: .primaryAction) {
             Button {
-              let _ = preview?.data?.saveAsFileToDownloads(filename: info.name)
+              if let fileURL = preview?.fileURL,
+                 let data = try? Data(contentsOf: fileURL) {
+                let _ = data.saveAsFileToDownloads(filename: info.name)
+              }
             } label: {
               Label("Save Text File...", systemImage: "square.and.arrow.down")
             }
@@ -100,7 +103,7 @@ struct FilePreviewTextView: View {
     }
     .task {
       if let info = info {
-        preview = FilePreview(info: info)
+        preview = FilePreviewState(info: info)
         preview?.download()
       }
     }