fixed bug for wordpress 4.0.1