Why do I get the following error when running code to draw DOS

[spe-or-feed]

File "/public/home/xuyk/.conda/envs/vaspvis/lib/python3.9/site-packages/vaspvis/dos.py", line 382, in _load_pdos
pdos = pdos[:, :, [(j * 4) + 4 for j in range(16)]]
IndexError: index 40 is out of bounds for axis 2 with size 37

[spe-or-feed]

Ps: I used dos.py normally when plotting 2D CrI3 with fewer atoms, but this error occurs when drawing 3D NdFe14B, is it because NdFeB has too many atoms?