Just a simple modification of my plugin