Fix to JAX upload shim fix, need to re-index the deps array.