Update tests

Update the tests to be compatible with commit ac00767.