Bug fix with menu loading