trying to fix the layout issue