Fixed Unit TestCase failure