stage_aware should be false if not set