fix layout issue in last commit