I'm definitely concerned that he accepted a fight with another wrestler after that mauling he took from Khabib. Good on him though. I don't think a 4 month layoff is too soon.
I don't really understand the rush for him, or how much progress expects to have achieved in a few months.
Really tough fight to come back to, and even a win won't really move him much closer to Khabib seeing how one sided that was.