small bug fix with the flag
small bug fixs with the flag all good now