Yes, you're correct. I've updated the code as per your suggestion.