Improve trait check

Fixes #83