Update the unit test to be more of a real world example.