fixes #4603

[erip]

trafficstars

Before submitting

• [x] Was this discussed/approved via a Github issue? (no need for typos, doc improvements)
• [x] Did you read the contributor guideline?
• [ ] Did you make sure to update the docs?
• [ ] Did you write any new necessary tests?

What does this PR do?

Fixes #4603.

PR review

Anyone in the community is free to review the PR once the tests have passed. If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[erip]

This failure looks unrelated...

[erip]

Any chance this could get merged @dianaml0? It's pretty small and is parallel to what happens in, e.g., metaseq so this seems like the right thing to do here.