Make test a little nicer