Fixing issue with layout height