Small fix for refactoring