Fixed revert to model