fixing error with layout