refactored 04 - logic in the test