update class to match WP core