Rewrite test for Minitest v5