Minor bug fix related to layout change.