[Fix] Fix the layout offset issue