Fix wrong jit source for atan2

Fixes another issue pointed out by @roystgnr in #366