Fix Unit Test