Replace `ensure` clause with setup block