Right now the module only works with nodes, expand it to support all entity types.