Fixed some bugs in new layout

Подчищаю за собой огрехи.