fix main layout, minor fix