Fixing edit layout