Fix Unit Test Assertion