One of Kemi Badenoch's top team has admitted there were flaws in the plan to return illegal migrants after Brexit, Sky News can reveal.