Fixed a bug where menu wouldn't work on return