fixed issue with layout