Small fix due to previous refactoring