remove dependancy to tf2