Forgot to remove some extra unused code on my last commit.