fixing test expectations