fix cb bug

fix cb bug