Update tests to latest version