Prosecutors in New York will no longer be able to seize a defendant’s assets in most cases unless they can show those funds are the direct product of illicit activity, according to legislation that’s expected to be passed as part of the state budget Sunday.

The proposal will amend the state’s law on so-called civil asset forfeiture, which is currently used by prosecutors in some cases to either freeze or collect the assets of a defendant throughout a criminal proceeding.