Improved code snippet to include WP Utils