Hacky fix for the the bug I inadvertently introduced in [1].

[1] 64abc7ef8589bfc87d9908513b43ec6f727b69fc