Fix check.py