Fix the collapse widget (thanks @zainiafzan)