Fix flowState layout display