Other small fix to layout