also use staging flags on input staging