Add missing expectation