Bug fix on wordpress plugin