diff --git a/nextaction.py b/nextaction.py
index b2d3b8a..3c8ca25 100755
--- a/nextaction.py
+++ b/nextaction.py
@@ -1,13 +1,15 @@
 #!/usr/bin/env python
 
-import time
 import logging
-import sys
 import argparse
-from datetime import datetime
 
+# noinspection PyPackageRequirements
 from todoist.api import TodoistAPI
 
+import time
+import sys
+from datetime import datetime
+
 
 def get_subitems(items, parent_item=None):
     """Search a flat item list for child items"""
diff --git a/setup.py b/setup.py
index 09ded72..ea2049e 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,4 @@
-from distutils.core import setup
+from setuptools import setup
 
 setup(
     name='NextAction',
@@ -10,8 +10,11 @@ setup(
     author_email='andy@tensixtyone.com',
     description='A more GTD-like workflow for Todoist. Uses the REST API to add and remove a @next_action label from tasks.',
     entry_points={
-        "distutils.commands": [
-            "nextaction = nextaction:main",
+        "console_scripts": [
+            "nextaction=nextaction:main",
             ],
-        }
+        },
+    install_requires=[
+        'todoist-python',
+    ]
 )