fix setup bug

fix setup bug