Change persistence module to remove of method