Updated unit test and it now passes.