fix($compile): updating the jQuery .context when it is replaced