Fix bug in previous commit.