Update the location of where we set legacy flags