fix flag bug