fixed numba issue, now compiles