Add test flag to avoid tests hanging when running for the first time (entering the setup mode).