diff --git a/attackcti/attack_api.py b/attackcti/attack_api.py
index a97e02d..2303158 100644
--- a/attackcti/attack_api.py
+++ b/attackcti/attack_api.py
@@ -11,7 +11,7 @@
 
 from stix2 import TAXIICollectionSource, Filter, CompositeDataSource, FileSystemSource
 from stix2.utils import get_type_from_id
-from taxii2client import Collection
+from taxii2client.v20 import Collection
 import json
 import os
 
diff --git a/setup.py b/setup.py
index d7ccc52..1227c5a 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@
 
 setuptools.setup(
     name="attackcti",
-    version="0.3.1",
+    version="0.3.2",
     author="Roberto Rodriguez",
     description="ATTACK CTI Libary",
     long_description=long_description,