Update tests again