Fix flag function