Modifie SimplePMG for tests