[Kwant] wave_function(...) returns array of size 0 when norbs=1