A federal judge granted Michael Cohen, President Donald Trump’s former personal lawyer, two additional months of freedom on Wednesday before he must surrender to federal prison.

Cohen pleaded guilty last year to charges that included campaign-finance violations for paying hush money to women who claimed to have had sex with Trump and for lying to Congress about the extent of negotiations for a Trump Tower in Russia. He was sentenced to three years in prison.

Cohen had been scheduled to report to prison March 6. U.S. District Judge William Pauley agreed Wednesday to delay that until May 6.