Modification for plugin registration