Fix type check
Merge pull request #24 from blom/fix-type-check

Fix type check