A federal judge on Thursday said he intended to order that Michael Cohen be returned to home confinement, after the former Trump attorney was sent back to prison earlier this month over a dispute with federal corrections officials.

Judge Alvin Hellerstein, of the U.S. District Court for the Southern District of New York, accused the federal Bureau of Prisons (BOP) of trying to violate Cohen's First Amendment rights by imposing a gag order as a condition of his home confinement.

"I make the finding that the purpose of transferring Mr. Cohen from furlough and home confinement to jail as retaliatory, and it's retaliatory because of his desire to exercise his First Amendment rights to publish the book and to discuss anything about the book or anything else he wants on social media" and elsewhere, Hellerstein said during a court hearing on Thursday.

The judge said that he wanted Cohen out of prison on Friday.

Cohen had been writing a book about his time working for President Trump and the ACLU filed a lawsuit this week alleging that he was sent back to prison in retaliation for the tell-all. Cohen is serving a three-year sentence for various charges, including fraud and lying to Congress.

“This order is a victory for the First Amendment,” Cohen's attorney Danya Perry said in a statement after the hearing. “The First Amendment does not allow the government to block Mr. Cohen from publishing a book critical of the president as a condition of his release to home confinement. This principle transcends politics. We are gratified that the rule of law prevails.”

Cohen had been released to home confinement in May amid concerns about the coronavirus pandemic's effects on the prison population.

He had pleaded guilty to the charges against him in 2018 and turned on Trump amid his legal ordeal, testifying before Congress and accusing the president of ordering him to make payments to two women who had claimed to have had affairs with the president.

Earlier this month, Cohen and his attorney met with corrections officials to finalize the terms of the home confinement agreement and objected to a number of the conditions, including a prohibition against speaking with the media or publishing any sort of writing.

Justice Department officials detained Cohen during the meeting over his objections and sent him back to prison.

The DOJ denied that the gag order was aimed at stopping Cohen from proceeding with his book or that his being returned to prison was retaliation over the planned publication.

During Thursday's hearing, Hellerstein, who was appointed to the court by former President Clinton, appeared disturbed by the manner in which BOP officials decided to reincarcerate Cohen and the gag order that they tried to impose upon him.

"I’ve never seen such a clause," Hellerstein said. "In 21 years of being a judge and sentencing people and looking at the conditions of supervised release, I’ve never seen such a clause."

A spokesman for the U.S. Attorney's Office for the Southern District of New York, which defended the BOP's move in court, declined a request for comment.

—Updated at 1:18 p.m.