fix update tests, forgot to add the code here