Fix flagging bug