Apply fix from commit 8f94d7d to duplicated method too

I forgot that you’d implemented the same method in two separate classes, here’s the same fix in the other class.