I’ve done this a few times. Even if you do 301 redirects properly, it still can take a month or two for google traffic to day get back to previous level. You might just have to be patient.