fixed issues with menu not resetting