Fixing layout error