Minor bug fix with displying of flags