Fix flag reference