Fixing problem with the way the unit test was written