Fixed broken code from last commit.