Fix compatibility issue with WP-Engine