fix the layout regression