Fix wp integration unit test