improved staging-custom-main test, and made it pass