fix bug

fix some bug by testing