Maybe use `ensure` here instead?