Change should into expect pattern