Fix tests for pg