this feels like a pretty strange place to put this, any reason why it can't go in `setup`?