Update.

Now works with wp 4.0