Fixes minor bug with menu selection