Fix last commit

MISP · Jan 18, 2017 · f593ce6 · f593ce6
1 parent 2bcc416
commit f593ce6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pymisp/api.py b/pymisp/api.py
@@ -370,20 +370,23 @@ def add_tag(self, event, tag, attribute=False):
         session = self.__prepare_session()
         if attribute:
             to_post = {'request': {'Attribute': {'id': event['id'], 'tag': tag}}}
+            path = 'attributes/addTag'
         else:
             to_post = {'request': {'Event': {'id': event['id'], 'tag': tag}}}
-        response = session.post(urljoin(self.root_url, 'events/addTag'), data=json.dumps(to_post))
+            path = 'events/addTag'
+        response = session.post(urljoin(self.root_url, path), data=json.dumps(to_post))
         return self._check_response(response)
 
     def remove_tag(self, event, tag, attribute=False):
         # FIXME: this is dirty, this function needs to be deprecated with something removing the tag to a UUID
         session = self.__prepare_session()
         if attribute:
             to_post = {'request': {'Attribute': {'id': event['id'], 'tag': tag}}}
-            pass
+            path = 'attributes/addTag'
         else:
             to_post = {'request': {'Event': {'id': event['Event']['id'], 'tag': tag}}}
-        response = session.post(urljoin(self.root_url, 'events/removeTag'), data=json.dumps(to_post))
+            path = 'events/addTag'
+        response = session.post(urljoin(self.root_url, path), data=json.dumps(to_post))
         return self._check_response(response)
 
     def _valid_uuid(self, uuid):