Fixed Unit test I broke