On Tuesday Amazon announced that they will stop testing jobseekers for marijuana.

The company, the second-largest private employer in the U.S. behind Walmart, is making the change as states legalize cannabis or introduce laws banning employers from testing for it.

Amazon said in a blog post that it will still test workers for other drugs and conduct “impairment checks” on the job. And the company said some roles may still require a cannabis test in line with Department of Transportation regulations.

Seattle-based Amazon also said Tuesday that it will support the federal legalization of marijuana by pushing lawmakers to pass the Marijuana Opportunity Reinvestment and Expungement Act of 2021.