Tidy up load/store jit

Tidy up