Fixing up a unit test to go along with a commit