update tests

thanks, making the change