small fix
improve wp