added test for editing checkpoint model