Fixes bug

Fixes bug where menu doesn't disappear