Correct bad import from numba