Use github.com/stretchr/testify for testing