Wrap expression with 'when' statement