HOTFIX: Slight tweak on previous commit.

Meant to merge this in when committing SPARK-3787.