Little fix.
Flags now stay flags after game over