nit: This function is pretty simple. Mind moving it into `run()` directly?