Minor tweak so I could make a pull request