Quickfix for the new WP

Fixed a compatibility bug