Fix bug introduced by recent change.