fix by wk 2

fix by wk 2