Fix callbacks for wp-3.9