Update version of tf2