Fixes

Minor fixes to check.py