Fix menu closing