Use testify for cleaner tests