Corrected small bug with Wordpress plugin