selftest: simplify logic in setup_env