Fix tests on PG