Fix buglet with Numba