fix by wk

fix by wk