Fixing a quick layout issue