nit: do this in the setup
(not mandatory but imo it's cleaner)