fixed to work with minitest 5