Fix flags if necessary