Missed off some of the refresh logic