Fix menu callback bug