From c6e53c9d2078c96ea2ac74d098fbc0db350dfa4f Mon Sep 17 00:00:00 2001
From: Darren <i@charact3.com>
Date: Fri, 1 Feb 2019 22:10:22 +0800
Subject: [PATCH] Update setup.py

---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index b7775cc..b27a071 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
 def get_version():
     regex = r"__version__\s=\s\'(?P<version>[\d\.ab]+?)\'"
 
-    path = 'aiocassandra.py'
+    path = 'aiocassandra/session.py'
 
     return re.search(regex, read(path)).group('version')
 
@@ -35,7 +35,7 @@ def read(*parts):
     setup_requires=['pytest-runner'] if needs_pytest else [],
     tests_require=['pytest', 'pytest-asyncio', 'pytest-cov'],
     python_requires='>=3.4.0',
-    py_modules=['aiocassandra'],
+    packages=['aiocassandra'],
     zip_safe=False,
     classifiers=[
         'Development Status :: 5 - Production/Stable',