fix & add flag