Fix bug in menu

Fixes #24