Yes, I forgot to update my code to do the correct thing. I'll write a utility method and also use it for line 77.