tweaking around, trying to find reason for assertion failures with `ao test.jit`