updated test for better coverage
WIP