Fix missing check.
Merge pull request #3 from scottso/fix_missing_check

Fix missing check.