Fixed up a unit test

Fixed a Unit test to do something.