simple refactoring

just tidying