fixing layout bug