Fix flex problem at input