Fix another bug in previous commit.