Fixed Failing Unit Test