fixing layout bug 2