fix expectation - same as in master