[BUGFIX] Make context menus work with modules