Fix WI generator script

[alexander-griffen]

Fixes #296

[alexander-griffen]

@zarino I've played with this locally and I'm fairly certain that it's the fix, but would you mind double checking on your system please?

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (e2a12cc) 82.50% compared to head (ae67f86) 82.50%.

Additional details and impacted files

@@           Coverage Diff           @@
##             main     #347   +/-   ##
=======================================
  Coverage   82.50%   82.50%           
=======================================
  Files          87       87           
  Lines        2669     2669           
  Branches      275      275           
=======================================
  Hits         2202     2202           
  Misses        404      404           
  Partials       63       63           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[zarino]

Annoyingly I can’t seem to replicate the issue in #296 on main, so either it was a transient error, or we’ve it’s been fixed by something else we’ve added in the meantime.

That said, this change seems fine to me, and the import seems to run for me, so I think we may as well merge this!