PR bug fixes