reworking the staging-custom-main test