Fix bug from last cleanup.