Refactoring
Small type change