make jit.py behave exactly like the binary