Updated the Unit test so that it accually passes.