update tests for my needs