fixed bug in display flex