From 280ec03ee928a65334ab30d67d20ea6cc7ec91c4 Mon Sep 17 00:00:00 2001
From: Justin Mazzocchi <2831158+jzzocc@users.noreply.github.com>
Date: Wed, 2 Sep 2020 19:23:58 -0700
Subject: [PATCH] Fix loading contexts of reblogs

---
 ViewModels/Sources/ViewModels/StatusListViewModel.swift | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/ViewModels/Sources/ViewModels/StatusListViewModel.swift b/ViewModels/Sources/ViewModels/StatusListViewModel.swift
index 3b18841..467b7c8 100644
--- a/ViewModels/Sources/ViewModels/StatusListViewModel.swift
+++ b/ViewModels/Sources/ViewModels/StatusListViewModel.swift
@@ -74,7 +74,9 @@ public extension StatusListViewModel {
     }
 
     func contextViewModel(id: String) -> StatusListViewModel {
-        StatusListViewModel(statusListService: statusListService.contextService(statusID: id))
+        let displayStatusID = statuses[id]?.displayStatus.id ?? id
+
+        return StatusListViewModel(statusListService: statusListService.contextService(statusID: displayStatusID))
     }
 }