This would also be nicer as a separate unit test.