Small refactor I noticed after checking in my last commit.