Improved integration with omnimenu.