Fixed a missing line in the last commit.