Fix slice issue with tfx (will do hotfix for release since it's nasty)