Correct instance variable and resotre functionality to site