DENVER (AP) — President Donald Trump has promised to support legislation protecting the marijuana industry in states that have legalized the drug, a move that could lift a threat to the industry made by the U.S. attorney general just three months ago.

Republican Sen. Cory Gardner of Colorado said Friday that Trump made the pledge to him in a Wednesday night conversation.

It marked the latest flip by the president who pledged while he was campaigning to respect states that legalized marijuana but also criticized legalization and implied it should be stopped.

Gardner has been pushing to reverse a decision made by Attorney General Jeff Sessions in January that removed prohibitions that kept federal prosecutors from pursuing cases against people who were following pot laws in states such as Colorado that have legalized the drug.

Marijuana has been fully legalized in eight states, and 24 states allow some form of marijuana use.

“President Trump has assured me that he will support a federalism-based legislative solution to fix this states’ rights issue once and for all,” Gardner said in a statement.

White House spokeswoman Sarah Huckabee Sanders said Gardner’s account was accurate and the president supported states’ rights in the matter.

Gardner hopes to introduce bipartisan legislation keeping the federal government from interfering in state marijuana markets.