update the test to be nicer
Merge pull request #2 from rylev/nicer-test

update the test to be nicer