forgot to remove the pdb.set_trace() call