I'll have a look soon. Perhaps @User can help with this since he made this feature.