Potty tainting

Is it okay to give your toddler the phone while they sit and do potty? I feel like that’s the only way it works for her to actually stay there for a bit