Fix site wrapper issue