Update tests to respect changes in pg module