Make staging test more stable, check memory while the staging is definitely running