Update cell state restoration api;