Fixed the Unit Test