Fix bug in the last commit.