From c1e345611de58fb09e4c467222d8b618a7a1e9a5 Mon Sep 17 00:00:00 2001
From: Thomas Wilshaw <thomaswilshaw@gmail.com>
Date: Sun, 27 Sep 2020 16:14:56 +0100
Subject: [PATCH] Cleanup.

---
 app/widget/projectexplorer/projectexplorer.cpp | 6 +++---
 app/widget/projectexplorer/projectexplorer.h   | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/app/widget/projectexplorer/projectexplorer.cpp b/app/widget/projectexplorer/projectexplorer.cpp
index 419f6a1ea..0694185e3 100644
--- a/app/widget/projectexplorer/projectexplorer.cpp
+++ b/app/widget/projectexplorer/projectexplorer.cpp
@@ -537,7 +537,7 @@ void ProjectExplorer::DeselectAll()
   CurrentView()->selectionModel()->clearSelection();
 }
 
-QList<Node*> ProjectExplorer::GetItemNodes(Item* item, Item::Type type) 
+QList<Node*> ProjectExplorer::GetFootageNodes(Item* item) 
 {
   // Output list list
   QList<Node*> nodes;
@@ -547,7 +547,7 @@ QList<Node*> ProjectExplorer::GetItemNodes(Item* item, Item::Type type)
   // Get item pointer.
   ItemPtr item_ptr = item->get_shared_ptr();
 
-  if (type == Item::kFootage) {
+  if (item_ptr.get()->type() == Item::kFootage) {
     // If no sequences exist we don't need to do anything clever here
     if (!sequences.isEmpty()) {
       // Footage can contain multiple streams, all of which need to be dealt with
@@ -621,7 +621,7 @@ void ProjectExplorer::DeleteSelected()
     if (item_ptr->type() == Item::kFootage) {
       
       // Check if nodes exists
-      QList<Node*> nodes = GetItemNodes(item, Item::kFootage);
+      QList<Node*> nodes = GetFootageNodes(item);
       if (!nodes.isEmpty()){
         FootageDeleteResponse response = DeleteWarningMessage();
         if (response == kOffline) {
diff --git a/app/widget/projectexplorer/projectexplorer.h b/app/widget/projectexplorer/projectexplorer.h
index 0e69b427c..12c999d5e 100644
--- a/app/widget/projectexplorer/projectexplorer.h
+++ b/app/widget/projectexplorer/projectexplorer.h
@@ -117,9 +117,9 @@ private:
 
   /**
    * @brief Check if an item is in use anywhere and return any relevant input nodes
-   * kFootage has two streams that need to be handled
+   * Returns a QList as Footage has two streams that need to be handled
    */
-  QList<Node*> GetItemNodes(Item* item, Item::Type type);
+  QList<Node*> GetFootageNodes(Item* item);
 
   /**
    * @brief Simple convenience function for adding a view to this stacked widget