Michael Cohen, the former personal attorney for President Donald Trump who is now a key witness in special counsel Robert Mueller’s Russia investigation, was under the impression Trump would offer him a pardon in exchange for staying on message in support of the President in discussions with federal prosecutors, according to two sources.

After a March 2018 visit to Mar-a-Lago, the President’s private club in Florida, Cohen returned to New York believing that his former boss would protect him if he faced any charges for sticking to his story about the 2016 payments to adult film actress Stormy Daniels, according to one source with knowledge. Trump was also at Mar-a-Lago at the time of Cohen’s visit.

Another source said that after the April 2018 FBI raid on Cohen’s office and home, people close to the President assured Cohen that Trump would take care of him. And Cohen believed that meant that the President would offer him a pardon if he stayed on message. It is unclear who specifically reached out to Cohen.