Update unit test to match PG fix.