Update tests

To comply with the changes in 5d0dc0c5a9e17bcb5fdcfe082024047730739cf2