Fix state restoration bug by correcting incorrectly named method. =/