diff --git a/setup.py b/setup.py
index afee25bac060ae019ff3381db6b1eb4b7e0641f0..e43515d9fc94adc33873c23d7492919b13ed0303 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,4 @@
 from setuptools import setup, find_packages
-import unittest
 from satella import __version__
 
 
@@ -15,7 +14,7 @@ setup(keywords=['ha', 'high availability', 'scalable', 'scalability', 'server',
             'HTTPJSONSource': ['requests'],
             'YAMLSource': ['pyyaml'],
             'TOMLSource': ['toml'],
-            'Faster JSONSource': ['ujson'],
-            'satella.cassandra': ['cassandra-driver']
+            'FasterJSONSource': ['ujson'],
+            'cassandra': ['cassandra-driver']
       }
       )