JIT: fix register preloading

Partially broken by typoes in the bitset patch.