update the test to be nicer