Fixes toggle button behaviour on small devices