Menu not collapsing - bug fixed