add import pdb at two places