fix persistance bug