update tests to make it ok