I think you mean restoreCallback here