forgot to add test file
refactoring