Update tests

Update tests to their latest versions