Fix bug that was left in previous commit.