diff --git a/Mastodon/Diffiable/Section/StatusSection.swift b/Mastodon/Diffiable/Section/StatusSection.swift
index e01276e8..cd4230da 100644
--- a/Mastodon/Diffiable/Section/StatusSection.swift
+++ b/Mastodon/Diffiable/Section/StatusSection.swift
@@ -273,8 +273,8 @@ extension StatusSection {
                 let statusTableViewCell = cell as! StatusTableViewCell
                 parent = statusTableViewCell.delegate?.parent()
                 playerViewControllerDelegate = statusTableViewCell.delegate?.playerViewControllerDelegate
-            case is NotificationTableViewCell:
-                let notificationTableViewCell = cell as! NotificationTableViewCell
+            case is NotificationStatusTableViewCell:
+                let notificationTableViewCell = cell as! NotificationStatusTableViewCell
                 parent = notificationTableViewCell.delegate?.parent()
             default:
                 parent = nil