Fixed problem with layout