Fix collapse button for mobile