Small fix to handle the vertical stacking