Size restore fix