Fix bug in the previous commit.