fix/update tests