Sorry, we should move the above logic into the `setup` function.