diff --git a/README.md b/README.md
index c42703e..8b95521 100644
--- a/README.md
+++ b/README.md
@@ -153,12 +153,9 @@ print(data == cbc_decrypted)  # True
 
 ## Testing
 
-1. Install the package: `pip3 install -U tgcrypto`.
 2. Clone this repository: `git clone https://github.com/pyrogram/tgcrypto`.
 3. Enter the directory: `cd tgcrypto`.
-4. Run tests: `python3 -m unittest`.
-
-**Note**: I haven't found any public test vectors for AES256-IGE, yet.
+4. Run tests: `python3 setup.py test`.
 
 ## License
 
diff --git a/setup.py b/setup.py
index a929860..aeb54c9 100644
--- a/setup.py
+++ b/setup.py
@@ -65,6 +65,7 @@ setup(
     },
     python_requires="~=3.4",
     packages=find_packages(),
+    test_suite="tests",
     zip_safe=False,
     ext_modules=[
         Extension(