Assiging now mostly works in the jitted code as well