Update source code for wordpress compatibility check