fixed injection issue into bundle_stage