Add update func to testenv