Fix bug in previous change