small refactor tests