Little code modification on add-on

Can be deleted please.