Fix bug when run one test