Fixes Index Out of Bounds

I'm only getting the failure on the CI, so it may fix - I don't see why it wouldn't fix...