Fixed bug - Sometimes special tags were not default when they should

1 files changed, 5 insertions, 7 deletions

diff --git a/service/app/tags.py b/service/app/tags.py
index 302e046c..7452b7d6 100644
--- a/service/app/tags.py
+++ b/service/app/tags.py
@@ -41,13 +41,11 @@ class Tags:
             self.tags[name] = self.add(name)
 
     def add(self, tag_input):
-        if tag_input.__class__.__name__ == 'str':
-            tag = Tag(tag_input, tag_input in self.SPECIAL_TAGS)
-            self.tags[tag_input] = tag
-            return tag
-        elif tag_input.__class__.__name__ == 'Tag':
-            self.tags[tag_input.name] = tag_input
-            return tag_input
+        if tag_input.__class__.__name__ == 'Tag':
+            tag_input = tag_input.name
+        tag = Tag(tag_input, tag_input in self.SPECIAL_TAGS)
+        self.tags[tag_input] = tag
+        return tag
 
     def find(self, name):
         return self.tags[name]