Set appropriate source module.

Closes #96.

Thanks to Matthew Flatt for the technique.
Make some more jit tests work.