small refactor
Merge pull request #1 from Rotsor/test

small refactor