add tests to check.sh sources list

2 files changed, 2 insertions, 2 deletions

diff --git a/check.sh b/check.sh
index e667d31..3b02b6d 100755
--- a/check.sh
+++ b/check.sh
@@ -2,7 +2,7 @@
 
 mypy -p pypaste || exit $?
 
-sources=($(find pypaste -type f))
+sources=($(find pypaste tests -type f -name '*.py'))
 
 flake8 ${sources[@]} || exit $?
 
diff --git a/tests/test_storage.py b/tests/test_storage.py
index e5f312c..7592920 100755
--- a/tests/test_storage.py
+++ b/tests/test_storage.py
@@ -7,7 +7,7 @@ import tempfile
 import aiosqlite
 import string
 import random
-from pypaste.server import Paste, Key, Storage, keygen
+from pypaste.server import Paste, Storage, keygen
 from pypaste.server.sqlite import Sqlite
 from pypaste.server.s3 import S3
 from datetime import datetime