Fix flag bug