added fix for menu collapse