Fix Matplotlib deprecation warning for ListedColormap

[cvanelteren]

We should fix

MatplotlibDeprecationWarning: Passing 'N' to ListedColormap is deprecated since 3.11 and will be removed in 3.13. Please ensure the list of passed colors is the required length instead