Fix state export driver

[duelingbenjos]

Description

There was some breakages in here along the way, this fixes this method which was originally intended for the state-export script.

Type of change

Please delete options that are not relevant.

• [x] Bug fix (non-breaking change which fixes an issue)