Fixed incorrect unit test assumptions