Fixed incorrect unit test