fix bug with layout