Fix typo introduced in refactor