Fix bug in previous change.