Small refactoring
Small change