diff --git a/py_ecc/bn128/bn128_curve.py b/py_ecc/bn128/bn128_curve.py
index 17938fac..0f6c8b07 100644
--- a/py_ecc/bn128/bn128_curve.py
+++ b/py_ecc/bn128/bn128_curve.py
@@ -6,7 +6,7 @@
     FQ2,
     FQ12,
 )
-from typing import Tuple, Union
+from typing import Tuple
 
 
 curve_order = 21888242871839275222246405745257275088548364400416034343698204186575808495617
diff --git a/py_ecc/bn128/bn128_field_elements.py b/py_ecc/bn128/bn128_field_elements.py
index 21300aca..d562bcc5 100644
--- a/py_ecc/bn128/bn128_field_elements.py
+++ b/py_ecc/bn128/bn128_field_elements.py
@@ -1,7 +1,6 @@
 from __future__ import absolute_import
 
 import sys
-from typing import NewType, Tuple
 from py_ecc.types import Point2D, Point3D
 
 sys.setrecursionlimit(10000)
diff --git a/py_ecc/secp256k1/secp256k1.py b/py_ecc/secp256k1/secp256k1.py
index 7314e195..1467143a 100644
--- a/py_ecc/secp256k1/secp256k1.py
+++ b/py_ecc/secp256k1/secp256k1.py
@@ -1,7 +1,6 @@
 import hashlib
 import hmac
 import sys
-from typing import NewType, Tuple, Union
 from py_ecc.types import Point2D, Point3D
 
 if sys.version_info.major == 2:
diff --git a/setup.py b/setup.py
index 836d81b0..e915005a 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,8 @@
 extras_require = {
     'test': [
         "pytest===3.2.2",
-        "pytest-xdist"
+        "pytest-xdist",
+        "mypy==0.641"
     ],
     'lint': [
         "flake8==3.4.1"