Fix broken test case on PG