Fixed Unit Test that was failing