diff --git a/requirements.txt b/requirements.txt
index 4bb6103..9fd7bd8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,3 +5,4 @@ scikit-learn >= 0.23.2
 tensorflow-gpu >= 1.15, < 2
 imutils >= 0.5.3
 matplotlib
+setuptools >= 50
diff --git a/setup.py b/setup.py
index a2cb66c..8e34b43 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,4 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-from setuptools import setup
+from setuptools import setup, find_packages
 
 install_requires = open('requirements.txt').read().split('\n')
 
@@ -12,7 +10,7 @@ setup(
     author='Vahid Rezanezhad',
     url='https://github.com/qurator-spk/eynollah',
     license='Apache License 2.0',
-    packages=['sbb_newspapers_org_image'],
+    packages=find_packages(),
     install_requires=install_requires,
     entry_points={
         'console_scripts': [