Use setup_module to simplify tests