Fix failed refactor in last commit.