Quick fix in update hook