Fixed outdated import
Merge update to use gopkg.in/check.v1