We only use `die` in one other place, afaik. `return` is correct here, I think.