Flag Bug

Flag Bug fix.